diff --git a/update_tags.py b/update_tags.py
index b1f4f428f..cb953b4bb 100644
--- a/update_tags.py
+++ b/update_tags.py
@@ -11,10 +11,13 @@
 
 if project in ["ttnn", "tt-metalium"]:
     version_no_v = version.replace("v", "")
+    #
+    # From version 0.54 forward, the wheel files were renamed to ttnn from metal_libs.
+    #
     additional_cmd = \
-        f"wget https://github.com/tenstorrent/tt-metal/releases/download/{version}/metal_libs-{version_no_v}+wormhole.b0-cp38-cp38-linux_x86_64.whl\n"
+        f"wget https://github.com/tenstorrent/tt-metal/releases/download/{version}/ttnn-{version_no_v}+wormhole.b0-cp38-cp38-linux_x86_64.whl\n"
     additional_cmd += \
-        f"pip install --extra-index-url https://download.pytorch.org/whl/cpu metal_libs-{version_no_v}+wormhole.b0-cp38-cp38-linux_x86_64.whl --force-reinstall"
+        f"pip install --extra-index-url https://download.pytorch.org/whl/cpu ttnn-{version_no_v}+wormhole.b0-cp38-cp38-linux_x86_64.whl --force-reinstall"
     version_structure = {version: {"additional_cmd": additional_cmd}}
 
 versions[project]["versions"].update(version_structure)