diff --git a/setup.py b/setup.py
index 27d9472..2e11002 100644
--- a/setup.py
+++ b/setup.py
@@ -1,8 +1,9 @@
 from setuptools import setup
+from setuptools import find_packages
 
 setup(name='gym_random_walk',
       version='0.0.1',
       install_requires=['gym'],
-      url="https://github.com/nczempin/gym-random-walk"
+      url="https://github.com/nczempin/gym-random-walk",
+      packages=find_packages()
 )  
-