diff --git a/paramnormal/tests/baseline_images/__init__.py b/paramnormal/tests/baseline_images/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/paramnormal/tests/baseline_images/test_activity/__init__.py b/paramnormal/tests/baseline_images/test_activity/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/setup.py b/setup.py
index 61aef0d..80821ff 100644
--- a/setup.py
+++ b/setup.py
@@ -30,7 +30,9 @@
     'Programming Language :: Python :: 3.5',
 ]
 INSTALL_REQUIRES = ['numpy', 'scipy', 'matplotlib']
-PACKAGE_DATA = {}
+PACKAGE_DATA = {
+    'paramnormal.tests.baseline_images.test_activity': ['*png'],
+}
 DATA_FILES = []
 
 
@@ -51,4 +53,5 @@
         platforms=PLATFORMS,
         classifiers=CLASSIFIERS,
         install_requires=INSTALL_REQUIRES,
+        zip_safe=False
     )