diff --git a/demo/multi-label-classification/multi_label_classifier.py b/demo/multi-label-classification/multi_label_classifier.py
index b1e7086f672e1a93a56c1baea94ec712b4761ef4..a3e38cb92569e930eed4b9cc88a8dba51143f11f 100644
--- a/demo/multi-label-classification/multi_label_classifier.py
+++ b/demo/multi-label-classification/multi_label_classifier.py
@@ -37,7 +37,7 @@ args = parser.parse_args()
 if __name__ == '__main__':
     # Load Paddlehub BERT pretrained model
     if args.use_taskid:
-        module = hub.Module(name="ernie_eng_base.hub_module")
+        module = hub.Module(name="ernie_v2_eng_base")
 
         inputs, outputs, program = module.context(
             trainable=True, max_seq_len=args.max_seq_len)