diff --git a/aimodel/src/lib/ai/model_rainfallwater_segmentation.py b/aimodel/src/lib/ai/model_rainfallwater_segmentation.py
index 404e995..551d992 100644
--- a/aimodel/src/lib/ai/model_rainfallwater_segmentation.py
+++ b/aimodel/src/lib/ai/model_rainfallwater_segmentation.py
@@ -55,7 +55,7 @@ def model_rainfallwater_segmentation(metadata, shape_water_out, model_arch="conv
 	model.compile(
 		optimizer="Adam",
 		loss=tf.keras.losses.SparseCategoricalCrossentropy(),
-		metrics=["accuracy"]
+		metrics=[tf.keras.losses.SparseCategoricalAccuracy()]
 	)
 	
 	return model
\ No newline at end of file