diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py
index b1844a1472..fc8b584caf 100644
--- a/unsloth/models/loader.py
+++ b/unsloth/models/loader.py
@@ -348,7 +348,7 @@ class FastLanguageModel(FastLlamaModel):
         #     dispatch_model = FastGraniteModel
         else:
             return FastModel.from_pretrained(
-                model_name                 = model_name,
+                model_name                 = old_model_name,
                 max_seq_length             = max_seq_length,
                 dtype                      = dtype,
                 load_in_4bit               = load_in_4bit,