diff --git a/setup.py b/setup.py
index 14a44250395538058d5c6dc49c7ab912808fc447..52ec6bdc3e28da0acdfc4251b360d18ec5a0c536 100644
--- a/setup.py
+++ b/setup.py
@@ -12,7 +12,7 @@ setuptools.setup(
     long_description=long_description,
     long_description_content_type="text/markdown",
     url="https://gitlab.fi.muni.cz/kypolab/theses/kosc-automated-problem-generation",
-    packages=setuptools.find_packages(),
+    packages=setuptools.find_namespace_packages(include=['generator'], exclude=['tests']),
     package_dir={'generator': './generator'},
     package_data={'generator': ['./*.txt', './*.yml']},
     classifiers=[