diff --git a/criteo_tft/requirements.txt b/criteo_tft/requirements.txt
index 67a6cb4fe..4e9e8b51c 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.7.2
 tensorflow-transform==0.4.0