diff --git a/projects/rocm-smi-lib/python_smi_tools/rocm_smi.py b/projects/rocm-smi-lib/python_smi_tools/rocm_smi.py
index 05cfd82645..c96d998f85 100755
--- a/projects/rocm-smi-lib/python_smi_tools/rocm_smi.py
+++ b/projects/rocm-smi-lib/python_smi_tools/rocm_smi.py
@@ -961,11 +961,11 @@ def resetGpu(device):
         logging.error('GPU Reset can only be performed on one GPU per call')
         RETCODE = 1
         return
-    if not isAmdDevice(device):
+    resetDev = int(device[0])
+    if not isAmdDevice(resetDev):
         logging.error('GPU Reset can only be performed on an AMD GPU')
         RETCODE = 1
         return
-    resetDev = int(device[0])
     ret = rocmsmi.rsmi_dev_gpu_reset(resetDev)
     if rsmi_ret_ok(ret, resetDev):
         printLog(resetDev, 'Successfully reset GPU %d' % (resetDev), None)