diff --git a/setup.py b/setup.py
index 09bd8cfe0321a7c1321b616400ea5a1f3d50f804..96afbe85dc99aa47533936d2a001b9d74717ec58 100644
--- a/setup.py
+++ b/setup.py
@@ -38,7 +38,7 @@ setup(
     include_package_data=True,
     keywords='flatland',
     name='flatland-rl',
-    packages=find_packages(include=['flatland']),
+    packages=find_packages('.'),
     setup_requires=setup_requirements,
     test_suite='tests',
     tests_require=test_requirements,