diff --git a/aimodel/src/encoderonly_test_rainfall.py b/aimodel/src/encoderonly_test_rainfall.py
index c6d4355..8e7a657 100755
--- a/aimodel/src/encoderonly_test_rainfall.py
+++ b/aimodel/src/encoderonly_test_rainfall.py
@@ -60,7 +60,7 @@ dataset_train, dataset_validate = dataset_encoderonly(
 # ██  ██  ██ ██    ██ ██   ██ ██      ██
 # ██      ██  ██████  ██████  ███████ ███████
 
-def make_encoderonly(windowsize, channels, encoder="convnext", water_bins=2):
+def make_encoderonly(windowsize, channels, encoder="convnext", water_bins=2, **kwargs):
 	if encoder == "convnext":
 		model = make_convnext(input_shape=(windowsize, windowsize, channels), num_classes=water_bins, **kwargs)
 	elif encoder == "resnet":