diff --git a/docs/Dockerfile b/docs/Dockerfile
index f168d7ca8a..5181177f0d 100644
--- a/docs/Dockerfile
+++ b/docs/Dockerfile
@@ -8,4 +8,4 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip
 RUN python3 -m pip install --no-cache-dir git+https://github.com/huggingface/doc-builder.git
 
 RUN git clone $clone_url && cd optimum && git checkout $commit_sha
-RUN python3 -m pip install --no-cache-dir ./optimum[onnxruntime,benchmark,quality,exporters-tf,exporters-executorch,doc-build,diffusers] 
+RUN python3 -m pip install --no-cache-dir ./optimum[onnxruntime,benchmark,quality,exporters-executorch,doc-build,diffusers]
diff --git a/setup.py b/setup.py
index 4ec891d7c1..caf0ae452f 100644
--- a/setup.py
+++ b/setup.py
@@ -75,7 +75,7 @@
         "transformers>=4.36,<4.47.0",
     ],
     "exporters-tf": [
-        # "tensorflow>=2.4,<=2.12.1",
+        "tensorflow>=2.4,<=2.12.1",
         "tf2onnx",
         "onnx",
         "onnxruntime",