diff --git a/setup.py b/setup.py
index ffb7796..38b639f 100644
--- a/setup.py
+++ b/setup.py
@@ -34,7 +34,7 @@
         "density-estimation",
         "normalizing-flow"
     ],
-    packages=["normalizing_flows"],
+    packages=find_packages(exclude=["test"]),
     python_requires=">=3.7, <4",
     install_requires=[
         "torch>=2.0.1",