diff --git a/OpenCOOD/opencood/visualization/simple_vis.py b/OpenCOOD/opencood/visualization/simple_vis.py
index 20f94131..0e8b3730 100644
--- a/OpenCOOD/opencood/visualization/simple_vis.py
+++ b/OpenCOOD/opencood/visualization/simple_vis.py
@@ -120,11 +120,4 @@ def visualize(
     else:
         raise (f"Not Completed for f{method} visualization.")
 
-    plt.axis("off")
-
-    plt.imshow(canvas.canvas)
-
-    plt.tight_layout()
-    plt.savefig(save_path, transparent=False, dpi=400, pad_inches=0.0)
-    plt.clf()
-    # print(save_path)
+    plt.imsave(save_path, canvas.canvas, dpi=400)
diff --git a/opencda/core/common/coperception_model_manager.py b/opencda/core/common/coperception_model_manager.py
index e4914214..e332c6ec 100644
--- a/opencda/core/common/coperception_model_manager.py
+++ b/opencda/core/common/coperception_model_manager.py
@@ -185,11 +185,8 @@ def render_inference_to_file(
             left_hand=left_hand,
             uncertainty=uncertainty,
         )
-        plt.axis("off")
-        plt.imshow(canvas)
-        plt.tight_layout()
-        plt.savefig(save_path, transparent=False, dpi=config["image_dpi"], pad_inches=0.0)
-        plt.clf()
+
+        plt.imsave(save_path, canvas, dpi=config["image_dpi"])
 
     @classmethod
     def visualize_inference_sample_dataloader(