diff --git a/.travis.yml b/.travis.yml
index 3fb748e..b21e852 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -1,8 +1,5 @@
 language: python
 python:
-  - "2.7"
-  - "3.4"
-  - "3.5"
   - "3.6"      # current default Python on Travis CI
   - "3.7"
   - "3.8"
@@ -12,7 +9,10 @@ os:
 # command to install dependencies
 install:
   - pip install -r requirements.txt
+  - pip install --upgrade pytest 
+  - pip install --upgrade pylint
+
 # command to run tests
 script:
-  - tests/test_dsm.py
+  - pytest tests/
   - pylint dsm/