diff --git a/.travis.yml b/.travis.yml
index 7e8ab0b..1b841af 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -1,5 +1,9 @@
 language: python
 
+branches:
+  only:
+    - master
+
 os:
   - linux
 
@@ -18,10 +22,12 @@ addons:
 
 install:
   - pip install --upgrade pip
-  - pip install coveralls pytest pytest-cov
+  - pip install --upgrade pytest
+  - pip install coveralls pytest-cov
 
 before_script:
   pip install -e .
+
 script:
   pytest --cov=numpythia tests/
 
diff --git a/setup.py b/setup.py
index a716f4f..69be39a 100755
--- a/setup.py
+++ b/setup.py
@@ -148,5 +148,6 @@ def finalize_options(self):
         'Development Status :: 5 - Production/Stable',
     ],
     install_requires=['numpy', 'six'],
+    tests_require=['pytest'],
     zip_safe=False,
 )