diff --git a/criteo_tft/requirements.txt b/criteo_tft/requirements.txt
index 67a6cb4fe..3eb9e4cb1 100644
--- a/criteo_tft/requirements.txt
+++ b/criteo_tft/requirements.txt
@@ -1,3 +1,3 @@
 six==1.10.0
-tensorflow==1.4.0
+tensorflow==2.5.1
 tensorflow-transform==0.4.0