diff --git a/fleetrec/demo/ctr-dnn_train.yaml b/fleetrec/demo/ctr-dnn_train.yaml
index 558fafcf631b42b9ddaf4dbc5cfafa6fe5fbcae4..37e1ab222f7d8df456ce5773923226de46d65e43 100644
--- a/fleetrec/demo/ctr-dnn_train.yaml
+++ b/fleetrec/demo/ctr-dnn_train.yaml
@@ -33,7 +33,7 @@ train:
     train_data_path: "fleetrec/models/ctr/dnn/data/train"
 
   model:
-    models: "fleetrec.models.ctr_dnn.model"
+    models: "fleetrec.models.ctr.dnn.model"
     hyper_parameters:
       sparse_inputs_slots: 27
       sparse_feature_number: 1000001
diff --git a/setup.py b/setup.py
index 1d6fabbdbb42ac8d0d33290e18b2160b90eddcb3..581832bf38a097151c95b186cf7d022243608346 100644
--- a/setup.py
+++ b/setup.py
@@ -27,7 +27,7 @@ setup(
     author=about["__author__"],
     author_email=about["__author_email__"],
     url=about["__url__"],
-    packages=find_packages("fleetrec"),
+    packages=find_packages(),
     python_requires=">=2.7",
     install_requires=requires,
     zip_safe=False