diff --git a/setup.py b/setup.py
index fa50856..c46f4c3 100644
--- a/setup.py
+++ b/setup.py
@@ -38,7 +38,7 @@
     "pylint >= 2.5.3",
     "pytz >= 2020.1",
     "python-dateutil",
-    "tensorflow == 2.3.1",
+    "tensorflow == 2.5.0",
     "tensorflow_transform >= 0.22",
 ]