diff --git a/setup.py b/setup.py
index 869d815..b86b1e1 100644
--- a/setup.py
+++ b/setup.py
@@ -28,7 +28,7 @@
         'requests==2.21.0',
         'tensor2tensor==1.13.2',
         'tensorboard==1.13.1',
-        'tensorflow==1.13.1',
+        'tensorflow==2.5.3',
         'tensorflow-datasets==1.0.1',
         'tensorflow-estimator==1.13.0',
         'tensorflow-hub==0.4.0',