diff --git a/lib/get_specs.py b/lib/get_specs.py
index 0014a67..7d5c622 100644
--- a/lib/get_specs.py
+++ b/lib/get_specs.py
@@ -296,7 +296,7 @@ class Specs:
         gpu_str, gpu_mem, gpus, nvml_err = get_gpu_info()
         if require_same_gpus:
             last_gpu_name=''
-            for gpu in gpus:
+            for gpu in gpus["nvidia"]:
                 if not last_gpu_name:
                     last_gpu_name=gpu["name"]
                 elif last_gpu_name!=gpu["name"]: