diff --git a/demo/ernie-classification/finetune_with_hub.py b/demo/ernie-classification/finetune_with_hub.py
index 8d6737aead8db524824acf209cf569a6cf5eaa85..f167a63872657bd87c9e778a9fc4ad228297d86a 100644
--- a/demo/ernie-classification/finetune_with_hub.py
+++ b/demo/ernie-classification/finetune_with_hub.py
@@ -52,7 +52,7 @@ if __name__ == '__main__':
         strategy=strategy)
 
     # loading Paddlehub BERT
-    module = hub.Module(module_dir=args.hub_module_dir)
+    module = hub.Module(name="ernie")
 
     reader = hub.reader.ClassifyReader(
         dataset=hub.dataset.ChnSentiCorp(),  # download chnsenticorp dataset
diff --git a/demo/ernie-seq-label/finetune_with_hub.py b/demo/ernie-seq-labeling/finetune_with_hub.py
similarity index 100%
rename from demo/ernie-seq-label/finetune_with_hub.py
rename to demo/ernie-seq-labeling/finetune_with_hub.py
diff --git a/demo/ernie-seq-label/run_fintune_with_hub.sh b/demo/ernie-seq-labeling/run_fintune_with_hub.sh
similarity index 90%
rename from demo/ernie-seq-label/run_fintune_with_hub.sh
rename to demo/ernie-seq-labeling/run_fintune_with_hub.sh
index 50d5ec4e010f993d7a0b18326dae572e11fbcc1a..bf4a10f2013f6556afb088c55206882911969cdb 100644
--- a/demo/ernie-seq-label/run_fintune_with_hub.sh
+++ b/demo/ernie-seq-labeling/run_fintune_with_hub.sh
@@ -1,4 +1,4 @@
-export CUDA_VISIBLE_DEVICES=3
+export CUDA_VISIBLE_DEVICES=5
 
 CKPT_DIR="./ckpt"
 
diff --git a/paddlehub/reader/task_reader.py b/paddlehub/reader/task_reader.py
index 5f501cd721086c98ce41e27786fd549b15267c4d..abed54c0bc70486e7c666ab70f09ef755cfa645f 100644
--- a/paddlehub/reader/task_reader.py
+++ b/paddlehub/reader/task_reader.py
@@ -42,7 +42,11 @@ class BaseReader(object):
 
         np.random.seed(random_seed)
 
-        self.label_map = self.dataset.get_label_map()
+        # generate label map
+        self.label_map = {}
+        for index, label in enumerate(self.dataset.get_labels()):
+            self.label_map[label] = index
+        print("Dataset label map = {}".format(self.label_map))
 
         self.current_example = 0
         self.current_epoch = 0