diff --git a/setup.py b/setup.py
index a399876..9823ccc 100755
--- a/setup.py
+++ b/setup.py
@@ -27,7 +27,7 @@
                     'statsmodels>=0.11.0',
                     'scipy',
                     'sklearn',
-                    'keras']
+                    'tensorflow']
 
 if __name__ == "__main__":
 
@@ -51,13 +51,13 @@
                     'wrangle.array'],
 
           classifiers=[
-                     'Intended Audience :: Science/Research',
-                     'Programming Language :: Python :: 3.6',
-                     'License :: OSI Approved :: MIT License',
-                     'Topic :: Scientific/Engineering :: Human Machine Interfaces',
-                     'Topic :: Scientific/Engineering :: Artificial Intelligence',
-                     'Topic :: Scientific/Engineering :: Mathematics',
-                     'Operating System :: POSIX',
-                     'Operating System :: Unix',
-                     'Operating System :: MacOS'],
-)
+              'Intended Audience :: Science/Research',
+              'Programming Language :: Python :: 3.6',
+              'License :: OSI Approved :: MIT License',
+              'Topic :: Scientific/Engineering :: Human Machine Interfaces',
+              'Topic :: Scientific/Engineering :: Artificial Intelligence',
+              'Topic :: Scientific/Engineering :: Mathematics',
+              'Operating System :: POSIX',
+              'Operating System :: Unix',
+              'Operating System :: MacOS'],
+          )
diff --git a/wrangle/utils/create_synth_model.py b/wrangle/utils/create_synth_model.py
index 97944e2..c91a1af 100644
--- a/wrangle/utils/create_synth_model.py
+++ b/wrangle/utils/create_synth_model.py
@@ -1,7 +1,7 @@
 def _base_for_model(mode, n=50, neurons=50):
 
-    from keras.models import Sequential
-    from keras.layers import Dense
+    from tensorflow.keras.models import Sequential
+    from tensorflow.keras.layers import Dense
 
     from .create_synth_data import create_synth_data