diff --git a/aimodel/src/lib/ai/RainfallWaterContraster.py b/aimodel/src/lib/ai/RainfallWaterContraster.py
index b452799..c871949 100644
--- a/aimodel/src/lib/ai/RainfallWaterContraster.py
+++ b/aimodel/src/lib/ai/RainfallWaterContraster.py
@@ -57,7 +57,7 @@ class RainfallWaterContraster(object):
 	
 	
 	def make_model(self):
-		model = model_rainfallwater_contrastive(batch_size=self.batch_size, **self.kwargs)
+		model = model_rainfallwater_contrastive(**self.kwargs)
 		return model