diff --git a/aimodel/src/subcommands/train_mono.py b/aimodel/src/subcommands/train_mono.py
index bc8a9da..bccbe03 100644
--- a/aimodel/src/subcommands/train_mono.py
+++ b/aimodel/src/subcommands/train_mono.py
@@ -54,8 +54,7 @@ def run(args):
 		dirpath_input=args.input,
 		batch_size=args.batch_size,
 		water_threshold=args.water_threshold,
-		shape_water_desired=[args.water_size, args.water_size],
-		dummy_label=False
+		shape_water_desired=[args.water_size, args.water_size]
 	)
 	dataset_metadata = read_metadata(args.input)