diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py
index f9b96fae4d..8ba7c45368 100644
--- a/unsloth/models/llama.py
+++ b/unsloth/models/llama.py
@@ -2644,7 +2644,7 @@ class FastLlamaModel:
     @staticmethod
     def for_training(model, use_gradient_checkpointing = True):
         if not hasattr(model, "parameters"):
-            raise TypeError("Unsloth: I think you're passing a tokenizer, not the model to for_inference!")
+            raise TypeError("Unsloth: I think you're passing a tokenizer, not the model to for_training!")
 
         # Delete all fast inference loras
         for param in model.parameters():