diff --git a/anaconda_verify/const.py b/anaconda_verify/const.py
index 4766335..989c78c 100644
--- a/anaconda_verify/const.py
+++ b/anaconda_verify/const.py
@@ -25,7 +25,7 @@
               'always_include_files'},
     'requirements': {'build', 'run'},
     'app': {'entry', 'icon', 'summary', 'type', 'cli_opts'},
-    'test': {'requires', 'commands', 'files', 'imports'},
+    'test': {'requires', 'commands', 'files', 'imports', 'source_files'},
     'about': {'license', 'license_url', 'license_family', 'license_file',
               'summary', 'description', 'home', 'doc_url', 'dev_url'},
 }
diff --git a/setup.py b/setup.py
index 9735b5d..6219556 100644
--- a/setup.py
+++ b/setup.py
@@ -2,8 +2,12 @@
 # All Rights Reserved
 import re
 from os.path import join
+import sys
 
-from distutils.core import setup
+if 'develop' in sys.argv:
+    from setuptools import setup
+else:
+    from distutils.core import setup
 
 
 # read version from anaconda_verify/__init__.py