diff --git a/setup.py b/setup.py
index cdd28210..199ef5ff 100644
--- a/setup.py
+++ b/setup.py
@@ -96,9 +96,7 @@ def build_assets(self):
         'Topic :: Scientific/Engineering :: Artificial Intelligence',
     ],
     cmdclass={'build_py': BuildPy},
-    package_dir={
-        'meltingpot': 'meltingpot',
-    },
+    packages=setuptools.find_packages(include=['meltingpot', 'meltingpot.*']),
     package_data={
         'meltingpot.lua': ['**'],
     },