diff --git a/.travis.yml b/.travis.yml
index 56e0da4..d7e53f1 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -8,8 +8,7 @@ env:
   - DJANGO_VERSION=1.11
 install:
   - pip install -q Django==$DJANGO_VERSION
-  - python setup.py -q install
-  - pip install .[flake8]
+  - pip install -e .[flake8,tests]
 script:
   - flake8
-  - ./run_tests.py
+  - pytest
diff --git a/setup.py b/setup.py
index 9de20a6..58c384e 100644
--- a/setup.py
+++ b/setup.py
@@ -29,6 +29,7 @@
         'isort==4.3.4',
         'testfixtures==6.3.0',
     ],
+    'tests': TESTS,
 }