diff --git a/fleetrec/core/factory.py b/fleetrec/core/factory.py
index 9a1fa3cf76f33cade87e6f6e91139ab06e558ff7..3aa3197deafc38c95f78a55c10961a1b1c3ee0dc 100644
--- a/fleetrec/core/factory.py
+++ b/fleetrec/core/factory.py
@@ -32,7 +32,7 @@ class TrainerFactory(object):
     def _build_trainer(config, yaml_path):
         print(envs.pretty_print_envs(envs.get_global_envs()))
 
-        train_mode = envs.get_global_env("train.strategy.mode")
+        train_mode = envs.get_global_env("train.trainer")
 
         if train_mode is None:
             train_mode = envs.get_runtime_envion("train.trainer")
diff --git a/fleetrec/examples/build_in/ctr-dnn_train.yaml b/fleetrec/examples/build_in/ctr-dnn_train.yaml
index a1c1eaa57a36a02947f72c336990e72a6a26e7c2..87de5fae0b41562f0efc0853767e39c0d0f07ae7 100644
--- a/fleetrec/examples/build_in/ctr-dnn_train.yaml
+++ b/fleetrec/examples/build_in/ctr-dnn_train.yaml
@@ -19,7 +19,7 @@ train:
   reader:
     mode: "dataset"
     batch_size: 2
-    class: "fleetrec.models.ctr_dnn.data_generator"
+    class: "fleetrec.models.ctr_dnn.reader"
     train_data_path: "/root/FleetRec/fleetrec/models/ctr_dnn/data/train/"
 
   model:
diff --git a/setup.py b/setup.py
index e1390ba5f7f46a575209bae6ebbb2e8a8ebf0fec..ca65bb3a64516368001cbaf7b9f30d074f7deeaf 100644
--- a/setup.py
+++ b/setup.py
@@ -4,7 +4,7 @@ setup for fleet-rec.
 
 from setuptools import setup
 
-packages = ["fleetrec", "fleetrec.models",
+packages = ["fleetrec", "fleetrec.models", "fleetrec.models.ctr_dnn",
             "fleetrec.examples", "fleetrec.examples.user_define", "fleetrec.examples.build_in",
             "fleetrec.core", "fleetrec.core.engine",
             "fleetrec.core.metrics", "fleetrec.core.models", "fleetrec.core.trainers", "fleetrec.core.utils"]