diff --git a/tools/keras_script.py b/tools/keras_script.py
index 04e0882902..d08379359d 100644
--- a/tools/keras_script.py
+++ b/tools/keras_script.py
@@ -3,7 +3,6 @@
 import json
 import os
 import pydoc
-import pathlib
 import re
 
 os.environ["KERAS_BACKEND"] = "jax"
@@ -170,11 +169,6 @@ def _update_headers(schema, docstring):
 
 def _metadata():
 
-    file = pathlib.Path('third_party/env/keras/lib/python3.12/site-packages/keras/__init__.py')
-    content = file.read_text(encoding='utf-8')
-    content = content.replace('from keras.api import _tf_keras', 'from keras import _tf_keras')
-    file.write_text(content, encoding='utf-8')
-
     root = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))
     json_path = os.path.join(root, 'source', 'keras-metadata.json')
     json_root = json.loads(_read(json_path))