diff --git a/setup.py b/setup.py
index 5b11d1df3..376f4df38 100644
--- a/setup.py
+++ b/setup.py
@@ -37,13 +37,14 @@ def compile_summary_protobuf():
 
 
 def build_package(version):
+    packages = setuptools.find_packages(include=["smdebug", "smdebug.*"])
     setuptools.setup(
         name="smdebug",
         version=version,
         author="AWS DeepLearning Team",
         description="Automated debugging for machine learning",
         url="https://github.com/awslabs/tornasole_core",
-        packages=setuptools.find_packages(),
+        packages=packages,
         classifiers=[
             "Programming Language :: Python :: 3",
             "Programming Language :: Python :: 3.6",