diff --git a/tftrt/benchmarking-python/benchmark_runner.py b/tftrt/benchmarking-python/benchmark_runner.py
index b02308665..32cc38ba6 100644
--- a/tftrt/benchmarking-python/benchmark_runner.py
+++ b/tftrt/benchmarking-python/benchmark_runner.py
@@ -75,7 +75,7 @@ def __init__(self, args):
         self._args = args
 
         # Temporary fix to re-enable NHWC layout.
-        os.environ["TF_ENABLE_LAYOUT_NHWC"] = "1"
+        # os.environ["TF_ENABLE_LAYOUT_NHWC"] = "1"
 
         if args.use_xla_auto_jit:
             logging.info("[Benchmark] - Activating XLA JIT Auto Clustering")