diff --git a/R/callbacks.R b/R/callbacks.R
index dc085285..e57f69e3 100644
--- a/R/callbacks.R
+++ b/R/callbacks.R
@@ -435,7 +435,7 @@ luz_callback_lr_scheduler <- luz_callback(
       lr_scheduler(optimizer, ...)
     }
     self[[call_on]] <- function() {
-      if ("metrics" %in% names(formals(scheduler$step))) {
+      if ("metrics" %in% names(formals(self$scheduler$step))) {
         self$scheduler$step(current_loss)
       } else {
         self$scheduler$step()