diff --git a/setup.py b/setup.py
index d73da5b..0132495 100644
--- a/setup.py
+++ b/setup.py
@@ -30,5 +30,5 @@
         'Intended Audience :: Developers',
         'Programming Language :: Python :: 3',
     ],
-    package_data={'models': ['models/*']},
+    package_data={'plasflow': ['models/*']},
 )