diff --git a/depyf/explain/utils.py b/depyf/explain/utils.py
index 4938fdcb..40fc6e10 100644
--- a/depyf/explain/utils.py
+++ b/depyf/explain/utils.py
@@ -99,6 +99,7 @@ def __init__(self, original_code, module, cache):
 
         try:
             klass = getattr(torch._dynamo.guards, "GuardManagerWrapper", None) or \
+                getattr(torch._dynamo.guards, "GuardManager", None) or \
                 getattr(torch._C._dynamo.guards, "GuardManager", None)
             assert klass is not None
             cpp_guard = isinstance(guard_manager, klass)