diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py
index 6ceda4f..6c302e2 100644
--- a/src/twomartens/masterthesis/cli.py
+++ b/src/twomartens/masterthesis/cli.py
@@ -358,7 +358,7 @@ def _ssd_test(args: argparse.Namespace) -> None:
                       ssd_model,
                       use_dropout,
                       forward_passes_per_image,
-                      image_size,
+                      (image_size, image_size),
                       output_path,
                       nr_digits)