diff --git a/src/utils/specs.py b/src/utils/specs.py
index 99979bd92d..38b2c780b5 100644
--- a/src/utils/specs.py
+++ b/src/utils/specs.py
@@ -224,7 +224,7 @@ def gpuinfo():
 
 def run(cmd):
     p = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-    if cmd[0] == "rocm-smi" and p.returncode == 8:
+    if cmd[0] == "rocm-smi" and p.returncode != 0:
         print("ERROR: No GPU detected. Unable to load rocm-smi")
         sys.exit(1)
     return p.stdout.decode("utf-8")