diff --git a/src/ptbench/engine/callbacks.py b/src/ptbench/engine/callbacks.py
index d730cf68c95e69fe4191dd5e6bdf95ed8b9ee596..20d457d10edd0c0feec554e84a2098156e7b728c 100644
--- a/src/ptbench/engine/callbacks.py
+++ b/src/ptbench/engine/callbacks.py
@@ -46,7 +46,7 @@ class LoggingCallback(Callback):
         self.log("total_time", current_time)
         self.log("eta", eta_seconds)
         self.log("loss", numpy.average(self.training_loss))
-        self.log("learning_rate", pl_module.lr)
+        self.log("learning_rate", pl_module.hparams["optimizer_params"]["lr"])
         self.log("validation_loss", numpy.average(self.validation_loss))
 
         queue_retries = 0