diff --git a/setup.py b/setup.py
index 2903dde8..88e412b0 100644
--- a/setup.py
+++ b/setup.py
@@ -57,7 +57,7 @@ def read_version():
                       'pandas', 'Pillow', 'h5py'],
     extras_require={
         'test': ['tox', 'flake8', 'pytest', 'pytest-cov', 'pytest-xdist', 'mock',
-                 'sagemaker==1.19.1', 'tensorflow', 'docker-compose', 'botocore>=1.12.140'],
+                 'sagemaker==1.19.1', 'tensorflow<2.0', 'docker-compose', 'botocore>=1.12.140'],
         'benchmark': ['click']
     },
 )