diff --git a/setup.py b/setup.py
index 5fb966c7..30d86561 100644
--- a/setup.py
+++ b/setup.py
@@ -42,7 +42,8 @@
     maintainer_email='support@datarobot.com',
     license='BSD',
     url='http://www.datarobot.com/',
-    packages=['datarobot_batch_scoring'],
+    packages=['datarobot_batch_scoring',
+              'datarobot_batch_scoring.network'],
     classifiers=[
         'License :: OSI Approved :: Apache Software License',
         'Programming Language :: Python :: 2.7',