diff --git a/src/mednet/scripts/upload.py b/src/mednet/scripts/upload.py
index e3a4ff75c86baa2ad34b4ccd8527e95ca8a192da..ea659e2246116b5e243632acde3585ce77e2de9d 100644
--- a/src/mednet/scripts/upload.py
+++ b/src/mednet/scripts/upload.py
@@ -187,9 +187,9 @@ def upload(
             mlflow.log_param(k, train_data[k])
 
         click.secho(
-            f"  -> `#accumulations` ({train_data['batch-chunk-count']})"
+            f"  -> `#accumulations` ({train_data['accumulate-grad-batches']})"
         )
-        mlflow.log_param("#Accumulations", train_data["batch-chunk-count"])
+        mlflow.log_param("#Accumulations", train_data["accumulate-grad-batches"])
         click.secho(f"  -> `epoch (best)` ({best_epoch})")
         mlflow.log_param("Epoch (best)", best_epoch)