diff --git a/setup.py b/setup.py
index 73abf74..0a45b24 100644
--- a/setup.py
+++ b/setup.py
@@ -27,7 +27,11 @@
        author='Hannes Bretschneider',
        author_email='hannes@psi.utoronto.ca',
        license='GPLv2',
-       packages=['hebel'],
+       packages=['hebel',
+                 'hebel.models',
+                 'hebel.layers',
+                 'hebel.utils',
+                 'hebel.pycuda_ops'],
        install_requires=[
            'pycuda',
            'numpy',
@@ -39,4 +43,4 @@
        scripts=['train_model.py'],
        include_package_data=True,
        zip_safe=False
-)
\ No newline at end of file
+)