diff --git a/setup.py b/setup.py
index 204ee2cc0..9cc707a87 100644
--- a/setup.py
+++ b/setup.py
@@ -1,7 +1,8 @@
-from setuptools import setup
+from setuptools import setup, find_packages
 
 setup(
     name="guided-diffusion",
     py_modules=["guided_diffusion"],
     install_requires=["blobfile>=1.0.5", "torch", "tqdm"],
+    packages=find_packages()
 )