diff --git a/debug/accuracy_tools/msprobe/pytorch/monitor/csv2tb.py b/debug/accuracy_tools/msprobe/pytorch/monitor/csv2tb.py
index a807f7e27987713f5c3f3488088239745b4aa0df..0f015dceb4b18f210bf522ec4e1bba1f3d777c85 100644
--- a/debug/accuracy_tools/msprobe/pytorch/monitor/csv2tb.py
+++ b/debug/accuracy_tools/msprobe/pytorch/monitor/csv2tb.py
@@ -79,7 +79,7 @@ def write_step(output_dirpath, parse_step_result, rank, data_type):
             for op, value in ops.items():
                 tag = f"{vpp_name}/{op}"
                 writer.add_scalar(tag, value, step)
-    writer.flush()
+    writer.close()
 
 
 @recursion_depth_decorator("update_dict", max_depth=50)
diff --git a/debug/accuracy_tools/msprobe/pytorch/parse_tool/lib/visualization.py b/debug/accuracy_tools/msprobe/pytorch/parse_tool/lib/visualization.py
index 5b53831b1c6fb9280dbad5621ee222baa2712225..5e54a45f948d6bc081e12f88467ba6c876c1cbc9 100644
--- a/debug/accuracy_tools/msprobe/pytorch/parse_tool/lib/visualization.py
+++ b/debug/accuracy_tools/msprobe/pytorch/parse_tool/lib/visualization.py
@@ -83,4 +83,3 @@ class Visualization:
                         self.util.log.info("\nStatistic Info:")
                         title_printed = True
                     self.util.log.info(summery_info)
-            pkl_handle.close()