diff --git a/.circleci/config.yml b/.circleci/config.yml
index 90327418..951c03d0 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -36,16 +36,4 @@ jobs:
           name: Test all the things!
           command: |
             . ~/venv/bin/activate
-            python3 test/test_batching.py
-            python3 test/test_bucket_scheduling.py
-            python3 test/test_distributed.py
-            python3 test/test_edgelist.py
-            python3 test/test_entitylist.py
-            python3 test/test_fileio.py
-            python3 test/test_functional.py
-            python3 test/test_losses.py
-            python3 test/test_model.py
-            python3 test/test_schema.py
-            python3 test/test_stats.py
-            python3 test/test_train.py
-            python3 test/test_util.py
+            python3 setup.py test
diff --git a/setup.cfg b/setup.cfg
index 517cea17..ac35ef50 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -33,6 +33,7 @@ keywords =
     knowledge-base
     graph-embedding
     link-prediction
+test_suite = test
 
 [options]
 setup_requires =
diff --git a/test/__init__.py b/test/__init__.py
new file mode 100644
index 00000000..e69de29b