diff --git a/aimodel/src/lib/ai/model_rainfallwater_contrastive.py b/aimodel/src/lib/ai/model_rainfallwater_contrastive.py
index f7e18de..ddcc83b 100644
--- a/aimodel/src/lib/ai/model_rainfallwater_contrastive.py
+++ b/aimodel/src/lib/ai/model_rainfallwater_contrastive.py
@@ -48,5 +48,5 @@ def model_rainfallwater_contrastive(shape_rainfall, shape_water, feature_dim=204
 	
 	model.compile(
 		optimizer="Adam",
-		loss=LossContrastive(weights_temperature=weight_temperature)
+		loss=LossContrastive(weight_temperature=weight_temperature)
 	)
\ No newline at end of file