diff --git a/setup.py b/setup.py
index 39a1b7ab8158a6c97e4bd41957f3da43d61f6849..79759a5e19dbff27f7d0fee31b57421a7ba8c698 100644
--- a/setup.py
+++ b/setup.py
@@ -4,9 +4,8 @@ setup for fleet-rec.
 
 from setuptools import setup
 
-packages = ["fleetrec", "fleetrec.examples", "fleetrec.metrics", "fleetrec.models", "fleetrec.models.ctr_dnn",
-            "fleetrec.reader",
-            "fleetrec.trainer", "fleetrec.utils"]
+packages = ["fleetrec", "fleetrec.models", "fleetrec.examples", "fleetrec.core", "fleetrec.core.engine",
+            "fleetrec.core.metrics", "fleetrec.core.models", "fleetrec.core.trainers", "fleetrec.core.utils"]
 
 requires = [
     "paddlepaddle"