diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py
index 2c5b0e9..8469034 100644
--- a/src/twomartens/masterthesis/cli.py
+++ b/src/twomartens/masterthesis/cli.py
@@ -341,7 +341,8 @@ def _ssd_test(args: argparse.Namespace) -> None:
                                 predictor_sizes=ssd_model.predictor_sizes,
                                 batch_size=batch_size,
                                 resized_shape=(image_size, image_size),
-                                training=False, evaluation=True, augment=False)
+                                training=False, evaluation=True, augment=False,
+                                nr_trajectories=1)
     del file_names_photos, instances
 
     nr_digits = math.ceil(math.log10(math.ceil(length_dataset / batch_size)))