diff --git a/unsloth/trainer.py b/unsloth/trainer.py
index 65abe6801f..8bb4440021 100644
--- a/unsloth/trainer.py
+++ b/unsloth/trainer.py
@@ -132,8 +132,8 @@ except:
 
 class UnslothTrainingArguments(TrainingArguments):
     def __init__(self, embedding_learning_rate: float = None, *args, **kwargs):
-        embedding_learning_rate = embedding_learning_rate
         super().__init__(*args, **kwargs)
+        self.embedding_learning_rate = embedding_learning_rate
 
 
 def _create_unsloth_optimizer(