diff --git a/connito/validator/run.py b/connito/validator/run.py
index ea4bd36..f4b773c 100644
--- a/connito/validator/run.py
+++ b/connito/validator/run.py
@@ -371,7 +371,6 @@ def setup_training(
             config=config,
             checkpoint_path=latest_checkpoint_path,
             outer_optimizer=outer_optimizer,
-            outer_scaler=outer_scaler,
             rank=rank,
             device=device,
             data_loader=train_dataloader,