diff --git a/aimodel/src/lib/vis/segmentation_plot.py b/aimodel/src/lib/vis/segmentation_plot.py
index a018f03..aca3165 100644
--- a/aimodel/src/lib/vis/segmentation_plot.py
+++ b/aimodel/src/lib/vis/segmentation_plot.py
@@ -25,4 +25,5 @@ def segmentation_plot(water_actual, water_predict, model_code, filepath_output):
 	
 	
 	plt.suptitle(f"Rainfall → Water depth prediction | {model_code}", fontsize=28, weight="bold")
-	plt.savefig(filepath_output)
\ No newline at end of file
+	plt.savefig(filepath_output)
+	plt.close()
\ No newline at end of file