diff --git a/setup.py b/setup.py
index 8de5c65..bae42f4 100644
--- a/setup.py
+++ b/setup.py
@@ -4,10 +4,7 @@
 import re
 
 
-requirements = ['numpy', 'pyyaml']
-#if os.environ.get('READTHEDOCS') != 'True':
-requirements.append('tensorflow')
-requirements.append('tensorflow-probability')
+requirements = ['numpy', 'pyyaml', 'tensorflow', 'tensorflow-probability']
 
 PACKAGE = 'pdfflow'