--- setup.py
+++ setup.py
@@ -70,6 +70,8 @@
         "enthought",
         "enthought.pyface",
         "enthought.pyface.ui",
+        "enthought.traits",
+        "enthought.traits.ui",
         ],
     package_data = {
         '': ['library/*', 'images/*'],