diff --git a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/overflow_check/overflow_check.py b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/overflow_check/overflow_check.py
index dc43c2c3bee01e036c3d6363dcbe4db8a085f590..a91a18a1e8447423d7b30e73bbdfc640ddcb3f09 100644
--- a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/overflow_check/overflow_check.py
+++ b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/overflow_check/overflow_check.py
@@ -86,7 +86,7 @@ def overflow_check(name, **kwargs):
     if not pid:
         return RuntimeError("Not get the specified process pid.")
 
-    def overflowcheck_hook(module, in_feat, out_feat):
+    def overflowcheck_hook(module, in_feat, out_feat=None):
         if not check_overflow_environment(pid):
             return
         dump_file = DumpUtil.get_dump_path()