diff --git a/eval/compare_q_transformers.py b/eval/compare_q_transformers.py
index 43194a0..959a0fa 100644
--- a/eval/compare_q_transformers.py
+++ b/eval/compare_q_transformers.py
@@ -17,14 +17,12 @@ except ModuleNotFoundError:
 
 try:
     from awq.modules.linear import WQLinear_GEMM
-except ModuleNotFoundError:
-    WQLinear_GEMM = None
-except ImportError:
+except (ModuleNotFoundError, ImportError):
     WQLinear_GEMM = None
 
 try:
     from vptq import VQuantLinear
-except ModuleNotFoundError:
+except (ModuleNotFoundError, ImportError):
     VQuantLinear = None
 
 try: