diff --git a/aimodel/src/encoderonly_test_rainfall.py b/aimodel/src/encoderonly_test_rainfall.py
index 0b180bf..2925222 100755
--- a/aimodel/src/encoderonly_test_rainfall.py
+++ b/aimodel/src/encoderonly_test_rainfall.py
@@ -67,6 +67,7 @@ def make_encoderonly(windowsize, channels, encoder="convnext", water_bins=2, **k
 			input_shape=(windowsize, windowsize, channels),
 			num_classes=water_bins,
 			downsample_at_start=False,
+			arch_name="convnext_xtiny",
 			**kwargs
 		)
 	elif encoder == "resnet":