diff --git a/gluefactory/train.py b/gluefactory/train.py
index 12ad207..debf212 100644
--- a/gluefactory/train.py
+++ b/gluefactory/train.py
@@ -331,7 +331,7 @@ def training(rank, conf, output_dir, args):
 
     results = None  # fix bug with it saving
 
-    lr_scheduler = get_lr_scheduler(optimizer=optimizer, conf=conf.train.lr_scheduler)
+    lr_scheduler = get_lr_scheduler(optimizer=optimizer, conf=conf.train.lr_schedule)
     if args.restore:
         optimizer.load_state_dict(init_cp["optimizer"])
         if "lr_scheduler" in init_cp: