diff --git a/setup.py b/setup.py
index 2cd84775..b0276c50 100644
--- a/setup.py
+++ b/setup.py
@@ -31,7 +31,7 @@
     license=about['__license__'],
     long_description=long_description,
     long_description_content_type="text/markdown",
-    packages=find_packages(),
+    packages=find_packages(include=['e2cnn']),
     python_requires='>3.7',
     keywords=[
         'pytorch',