diff --git a/aimodel/src/lib/ai/model_rainfallwater_mono.py b/aimodel/src/lib/ai/model_rainfallwater_mono.py
index c3cf25a..3c546a0 100644
--- a/aimodel/src/lib/ai/model_rainfallwater_mono.py
+++ b/aimodel/src/lib/ai/model_rainfallwater_mono.py
@@ -22,6 +22,7 @@ def model_rainfallwater_mono(metadata, shape_water_out, model_arch_enc="convnext
 	"""
 	rainfall_channels, rainfall_width, rainfall_height = metadata["rainfallradar"] # shape = [channels, width, height]
 	
+	print("RAINFALL channels", rainfall_channels, "width", rainfall_width, "height", rainfall_height)
 	out_water_width, out_water_height = shape_water_out
 	
 	layer_input = tf.keras.layers.Input(
diff --git a/aimodel/src/lib/dataset/dataset_mono.py b/aimodel/src/lib/dataset/dataset_mono.py
index d9afc5a..7fc7292 100644
--- a/aimodel/src/lib/dataset/dataset_mono.py
+++ b/aimodel/src/lib/dataset/dataset_mono.py
@@ -19,6 +19,8 @@ def parse_item(metadata, shape_water_desired=[100,100], water_threshold=0.1, wat
 	water_width_target, water_height_target = shape_water_desired
 	water_offset_x = math.ceil((water_width_source - water_width_target) / 2)
 	water_offset_y = math.ceil((water_height_source - water_height_target) / 2)
+	
+	rainfall_channels, rainfall_width, rainfall_height = metadata["rainfallradar"]
 	def parse_item_inner(item):
 		parsed = tf.io.parse_single_example(item, features={
 			"rainfallradar": tf.io.FixedLenFeature([], tf.string),