diff --git a/pip_build.py b/pip_build.py
index 52cc0874743..6570ab8eb22 100644
--- a/pip_build.py
+++ b/pip_build.py
@@ -86,9 +86,8 @@ def build(root_path, is_nightly=False, rc_index=None):
         move_tf_keras_directory()
         print(os.getcwd())
 
-        # from keras.src.version import __version__  # noqa: E402
+        from keras.src.version import __version__  # noqa: E402
 
-        __version__ = "3.3.2"
         export_version_string(__version__, is_nightly, rc_index)
         return build_and_save_output(root_path, __version__)
     finally: