diff --git a/setup.py b/setup.py
index ce051120..32387d76 100755
--- a/setup.py
+++ b/setup.py
@@ -24,10 +24,12 @@
 except ImportError:
     from distutils.core import setup
 
-install_requires = ['wrangle',
+install_requires = ['statsmodels>=0.11.0',
+                    'wrangle',
                     'numpy',
                     'pandas',
-                    'keras',
+                    'tensorflow<=1.5.0',
+                    'keras==2.3.0',
                     'astetik',
                     'sklearn',
                     'tqdm',