diff --git a/debug/accuracy_tools/msprobe/pytorch/hook_module/script_wrapper.py b/debug/accuracy_tools/msprobe/pytorch/hook_module/script_wrapper.py
index 7d4c3176dbffcf0ecb57094e5a817bf01717a8d5..136f0c9974527d433c995fe7e670c45913c6171c 100644
--- a/debug/accuracy_tools/msprobe/pytorch/hook_module/script_wrapper.py
+++ b/debug/accuracy_tools/msprobe/pytorch/hook_module/script_wrapper.py
@@ -63,7 +63,8 @@ def wrap_compile_script_func():
                                                                               hooks)  # type: ignore[attr-defined]
         return _wrapped
 
-    torch._dynamo.convert_frame = _patched_convert_frame
+    import torch._dynamo.convert_frame as _cf_mod
+    _cf_mod.convert_frame = _patched_convert_frame
 
 
 def wrap_script_func():