diff --git a/lib/utils.py b/lib/utils.py
index 3afaa67ed1f2607f20a15d22c931a905911795da..070bbad08180c93ded8c70f6f2b6c1c545c4f99f 100644
--- a/lib/utils.py
+++ b/lib/utils.py
@@ -649,18 +649,16 @@ def save_device_info(comp_insts, dstree, args, logdir):
                 continue
             modules.append(mod)
         driver_info = {}
-        for drv in dstree.drivers:
-            drv = drv.strip()
-            if drv in modules:
-                _, output = run_local_cmd('modinfo %s' % drv, shell=True)
-                info = {drv: {}}
-                for line in output.split('\n'):
-                    if ':' not in line:
-                        continue
-                    key, val = line.split(':', 1)
-                    val = val.strip()
-                    info[drv].update({key: val})
-                driver_info.update(info)
+        for drv in modules:
+            _, output = run_local_cmd('modinfo %s' % drv, shell=True)
+            info = {drv: {}}
+            for line in output.split('\n'):
+                if ':' not in line:
+                    continue
+                key, val = line.split(':', 1)
+                val = val.strip()
+                info[drv].update({key: val})
+            driver_info.update(info)
         return driver_info
 
     all_info = {}