diff --git a/setup.py b/setup.py
index e300c77911..4b9cf9e8a4 100644
--- a/setup.py
+++ b/setup.py
@@ -60,7 +60,7 @@ def read_version():
       install_requires=required_packages,
 
       extras_require={
-          'test': ['tox', 'flake8', 'pytest', 'pytest-cov', 'pytest-rerunfailures',
+          'test': ['tox', 'flake8', 'pytest==4.4.1', 'pytest-cov', 'pytest-rerunfailures',
                    'pytest-xdist', 'mock', 'tensorflow>=1.3.0', 'contextlib2',
                    'awslogs', 'pandas']},