diff --git a/debug/accuracy_tools/msprobe/core/common/file_utils.py b/debug/accuracy_tools/msprobe/core/common/file_utils.py
index 0976323b5a760c1e6d250923ab7bfdbc166a0bef..65163f3fa3fedebc82d0dbc79bdfd1defd12cf0d 100644
--- a/debug/accuracy_tools/msprobe/core/common/file_utils.py
+++ b/debug/accuracy_tools/msprobe/core/common/file_utils.py
@@ -241,7 +241,7 @@ def make_dir(dir_path):
     if os.path.isdir(dir_path):
         return
     try:
-        os.mkdir(dir_path, mode=FileCheckConst.DATA_DIR_AUTHORITY)
+        os.makedirs(dir_path, mode=FileCheckConst.DATA_DIR_AUTHORITY, exist_ok=True)
     except OSError as ex:
         raise FileCheckException(FileCheckException.ILLEGAL_PATH_ERROR,
                                  f"Failed to create {dir_path}. "
diff --git a/debug/accuracy_tools/msprobe/pytorch/hook_module/hook_module.py b/debug/accuracy_tools/msprobe/pytorch/hook_module/hook_module.py
index aa724b50fd43e6699f9cd1e5fb8b3d1efe59c778..c2feff0bdb51c1c1a7ae8a921832dbdf83d489d0 100644
--- a/debug/accuracy_tools/msprobe/pytorch/hook_module/hook_module.py
+++ b/debug/accuracy_tools/msprobe/pytorch/hook_module/hook_module.py
@@ -56,7 +56,7 @@ class HOOKModule(nn.Module):
             else:
                 self.register_forward_pre_hook(forward_pre_hook)
                 self.register_forward_hook(forward_hook)
-            self.register_backward_hook(backward_hook)
+            self.register_full_backward_hook(backward_hook)
 
     def __call__(self, *input, **kwargs):
         changed = False