diff --git a/basic_pitch/train.py b/basic_pitch/train.py
index 4d675fb..63fd7f4 100644
--- a/basic_pitch/train.py
+++ b/basic_pitch/train.py
@@ -268,8 +268,8 @@ def console_entry_point() -> None:
     main(
         args.source,
         args.output,
-        args.training_shuffle_size,
         args.batch_size,
+        args.training_shuffle_size,
         args.learning_rate,
         args.epochs,
         args.steps_per_epoch,