diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py
index 4e97cdc..db89590 100644
--- a/src/twomartens/masterthesis/cli.py
+++ b/src/twomartens/masterthesis/cli.py
@@ -242,7 +242,7 @@ def _ssd_val(args: argparse.Namespace) -> None:
     use_dropout = False if args.network == "ssd" else True
     
     weights_file = f"{args.weights_path}/VGG_coco_SSD_300x300_iter_400000.h5"
-    checkpoint_path = f"{args.weights_path}/train/{args.network}/"
+    checkpoint_path = f"{args.weights_path}/train/{args.network}/{args.train_iteration}"
     output_path = f"{args.output_path}/val/{args.network}/{args.iteration}/"
     os.makedirs(output_path, exist_ok=True)
     
diff --git a/src/twomartens/masterthesis/main.py b/src/twomartens/masterthesis/main.py
index 5afb272..11a0732 100644
--- a/src/twomartens/masterthesis/main.py
+++ b/src/twomartens/masterthesis/main.py
@@ -121,6 +121,7 @@ def _build_ssd_val(parser: argparse.ArgumentParser) -> None:
     parser.add_argument("--summary_path", type=str, help="path to the summaries directory")
     parser.add_argument("--output_path", type=str, help="path to the output directory")
     parser.add_argument("iteration", type=int, help="the validation iteration")
+    parser.add_argument("train_iteration", type=int, help="the train iteration")
     
 
 def _build_auto_encoder_val(parser: argparse.ArgumentParser) -> None: