diff --git a/python_smi_tools/rocm_smi.py b/python_smi_tools/rocm_smi.py
index 056ab2fb..0271e40e 100755
--- a/python_smi_tools/rocm_smi.py
+++ b/python_smi_tools/rocm_smi.py
@@ -66,7 +66,7 @@ def driverInitialized():
     """
     driverInitialized = ''
     try:
-        driverInitialized = str(subprocess.check_output("cat /proc/modules|grep amdgpu", shell=True))
+        driverInitialized = str(subprocess.check_output("cat /sys/module/amdgpu/initstate |grep live", shell=True))
     except subprocess.CalledProcessError:
         pass
     if len(driverInitialized) > 0: