diff --git a/aimodel/src/lib/ai/model_rainfallwater_segmentation.py b/aimodel/src/lib/ai/model_rainfallwater_segmentation.py
index 126fd73..87b4bab 100644
--- a/aimodel/src/lib/ai/model_rainfallwater_segmentation.py
+++ b/aimodel/src/lib/ai/model_rainfallwater_segmentation.py
@@ -31,7 +31,7 @@ def model_rainfallwater_segmentation(metadata, feature_dim_in, shape_water_out,
 	
 	model.compile(
 		optimizer="Adam",
-		loss=tf.keras.losses.SparseCategoricalCrossentropy()
+		loss=tf.keras.losses.SparseCategoricalCrossentropy(),
 		metrics=["accuracy"]
 	)