diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py
index 2341500..8343a46 100644
--- a/src/twomartens/masterthesis/cli.py
+++ b/src/twomartens/masterthesis/cli.py
@@ -62,7 +62,7 @@ def _ssd_train(args: argparse.Namespace) -> None:
     
     scenenet_data, nr_digits, length_dataset = \
         data.load_scenenet_data(file_names_photos, instances, args.coco_path,
-                                batch_size=batch_size,
+                                batch_size=batch_size, num_epochs=args.num_epochs,
                                 resized_shape=(image_size, image_size),
                                 mode="training")
     del file_names_photos, instances