diff --git a/setup.py b/setup.py
index 4de4e45..cc8cc2a 100644
--- a/setup.py
+++ b/setup.py
@@ -35,7 +35,7 @@ setup(
     package_data={},
     entry_points={},
     python_requires="~=3.6",
-    install_requires=["keras"],
+    install_requires=["tensorflow"],
     license="Apache License 2.0",
     classifiers=[
         "Operating System :: OS Independent",