diff --git a/aimodel/src/subcommands/rainfall_stats.py b/aimodel/src/subcommands/rainfall_stats.py
index ec19949..c856272 100755
--- a/aimodel/src/subcommands/rainfall_stats.py
+++ b/aimodel/src/subcommands/rainfall_stats.py
@@ -61,7 +61,7 @@ def run(args):
 		batch_stddev = tf.math.reduce_std(rainfall_flat)
 		batch_max = tf.math.reduce_max(rainfall_flat)
 		
-		print("BATCH mean\t", batch_mean.numpy().tolist(), "\tstddev\t", batch_stddev.numpy().tolist(), "\tmax\t", batch_max.numpy().tolist())
+		print("BATCH", i, "\tmean\t", batch_mean.numpy().tolist(), "\tstddev\t", batch_stddev.numpy().tolist(), "\tmax\t", batch_max.numpy().tolist())
 		
 		calc_mean.append(batch_mean)
 		calc_stddev.append(batch_stddev)