diff --git a/aimodel/src/lib/ai/model_rainfallwater_mono.py b/aimodel/src/lib/ai/model_rainfallwater_mono.py
index 1ceea01..9b15228 100644
--- a/aimodel/src/lib/ai/model_rainfallwater_mono.py
+++ b/aimodel/src/lib/ai/model_rainfallwater_mono.py
@@ -72,7 +72,7 @@ def model_rainfallwater_mono(metadata, shape_water_out, model_arch_enc="convnext
 	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