diff --git a/mlc_llm/build.py b/mlc_llm/build.py
index c90da542b8..b7619aa963 100644
--- a/mlc_llm/build.py
+++ b/mlc_llm/build.py
@@ -10,7 +10,8 @@ def debug_on_except():
     try:
         yield
     finally:
-        if sys.exc_info() == (None, None, None):
+        raised_exception = sys.exc_info()[1]
+        if not isinstance(raised_exception, Exception):
             return
 
         import traceback