diff --git a/official/requirements.txt b/official/requirements.txt
index 5f5838812fb..02d91b03f26 100644
--- a/official/requirements.txt
+++ b/official/requirements.txt
@@ -10,7 +10,7 @@ scipy>=0.19.1
 tensorflow-hub>=0.6.0
 tensorflow-model-optimization>=0.4.1
 tensorflow-datasets
-tf-keras
+tf-keras>=2.16.0
 gin-config
 tf_slim>=1.1.0
 Cython