diff --git a/python_smi_tools/rocm_smi.py b/python_smi_tools/rocm_smi.py
index ac4e02b2f4..15df5e30d5 100755
--- a/python_smi_tools/rocm_smi.py
+++ b/python_smi_tools/rocm_smi.py
@@ -395,12 +395,11 @@ def getTargetGfxVersion(device, silent=False):
         (you plan to handle manually). Default is off.
     """
     target_graphics_version = c_uint64()
-    market_name = str(getDeviceName(device, True))
     gfx_ver_ret = "N/A"
     ret = rocmsmi.rsmi_dev_target_graphics_version_get(device, byref(target_graphics_version))
     target_graphics_version = str(target_graphics_version.value)
     if rsmi_ret_ok(ret, device, 'get_target_gfx_version', silent=silent):
-        if len(target_graphics_version) == 4 and ("Instinct MI2" in market_name):
+        if target_graphics_version == "9010":
             hex_part = str(hex(int(str(target_graphics_version)[2:]))).replace("0x", "")
             target_graphics_version = str(target_graphics_version)[:2] + hex_part
         gfx_ver_ret = "gfx" + str(target_graphics_version)