diff --git a/backtesting/test/__main__.py b/backtesting/test/__main__.py
new file mode 100644
index 0000000..89c8911
--- /dev/null
+++ b/backtesting/test/__main__.py
@@ -0,0 +1,6 @@
+import unittest
+
+suite = unittest.defaultTestLoader.discover('backtesting.test',
+                                            pattern='_test*.py')
+if __name__ == '__main__':
+    unittest.TextTestRunner().run(suite)
diff --git a/backtesting/test/_test.py b/backtesting/test/_test.py
index c59b187..3fb34d2 100644
--- a/backtesting/test/_test.py
+++ b/backtesting/test/_test.py
@@ -500,6 +500,10 @@ class TestUtil(TestCase):
             self.assertEqual(_as_str(_Array([1], name=s)), s[0])
 
 
+@unittest.skipUnless(
+    os.path.isdir(os.path.join(os.path.dirname(__file__),
+                               '..', '..', 'doc')),
+    "docs dir doesn't exist")
 class TestDocs(TestCase):
     def test_examples(self):
         examples = glob(os.path.join(os.path.dirname(__file__),
diff --git a/setup.py b/setup.py
index f6cb477..d2f8819 100644
--- a/setup.py
+++ b/setup.py
@@ -5,13 +5,6 @@ if sys.version_info < (3, 4):
     sys.exit('ERROR: Backtesting.py requires Python 3.4+')
 
 
-def _discover_tests():
-    import unittest
-    return unittest.defaultTestLoader.discover('backtesting.test',
-                                               pattern='*test*.py',
-                                               top_level_dir='.')
-
-
 if __name__ == '__main__':
     from setuptools import setup, find_packages
 
@@ -52,7 +45,7 @@ if __name__ == '__main__':
                 'flake8',
             ],
         },
-        test_suite="setup._discover_tests",
+        test_suite="backtesting.test",
         python_requires='>=3.4',
         author='Zach Lûster',
         classifiers=[