diff --git a/docker/consistency.dockerfile b/docker/consistency.dockerfile
index 60769317a5734c..03740342c6fdbb 100644
--- a/docker/consistency.dockerfile
+++ b/docker/consistency.dockerfile
@@ -6,7 +6,7 @@ ENV VIRTUAL_ENV=/usr/local
 RUN pip install uv
 RUN uv venv
 RUN uv pip install --no-cache-dir -U pip setuptools
-RUN uv pip install --no-cache-dir tensorflow-cpu
+RUN uv pip install --no-cache-dir tensorflow-cpu tf-keras
 RUN uv pip install --no-cache --upgrade 'torch' --index-url https://download.pytorch.org/whl/cpu
 RUN pip install --no-cache-dir --upgrade GitPython "transformers[flax,quality]"
 RUN pip uninstall -y transformers