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