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