diff --git a/setup.py b/setup.py
index 25123b2c..662ba6de 100644
--- a/setup.py
+++ b/setup.py
@@ -27,8 +27,7 @@ def extra_cmake_args():
         'Programming Language :: Python :: 3',
     ],
     keywords='',
-    packages=['stempy', 'stempy.io', 'stempy.image', 'stempy.pipeline'],
-    package_dir={'':'python'},
+    packages=['stempy'],
     install_requires=[
         'numpy',
         'h5py',