diff --git a/src/mednet/scripts/experiment.py b/src/mednet/scripts/experiment.py
index 8cb233f0fbbdce0bde078eaf13579650931fc93b..ad4de8bcd7eff7684eb79cfcec0e963f33d35492 100644
--- a/src/mednet/scripts/experiment.py
+++ b/src/mednet/scripts/experiment.py
@@ -49,6 +49,7 @@ def experiment(
     parallel,
     monitoring_interval,
     balance_classes,
+    augmentations,
     **_,
 ):  # numpydoc ignore=PR01
     r"""Run a complete experiment, from training, to prediction and evaluation.
@@ -89,6 +90,7 @@ def experiment(
         parallel=parallel,
         monitoring_interval=monitoring_interval,
         balance_classes=balance_classes,
+        augmentations=augmentations,
     )
     train_stop_timestamp = datetime.now()