diff --git a/setup.py b/setup.py
index 3b30a377d..32a20fa16 100644
--- a/setup.py
+++ b/setup.py
@@ -147,7 +147,7 @@ def run(self):
       install_requires=install_requires,
       dependency_links=dep_links,
       test_requires=test_requires,
-      packages=['pyop2', 'pyop2.codegen'],
+      packages=['pyop2', 'pyop2.codegen', 'pyop2.types'],
       package_data={
           'pyop2': ['assets/*', '*.h', '*.pxd', '*.pyx', 'codegen/c/*.c']},
       scripts=glob('scripts/*'),