diff --git a/unsloth/models/mapper.py b/unsloth/models/mapper.py
index bc01c28583..6e6e402a09 100644
--- a/unsloth/models/mapper.py
+++ b/unsloth/models/mapper.py
@@ -555,12 +555,12 @@ __INT_TO_FLOAT_MAPPER = \
         "deepseek-ai/DeepSeek-R1-Distill-Llama-70B",
     ),
     "unsloth/Mistral-Small-24B-Base-2501-unsloth-bnb-4bit" : (
-        "unsloth/Mistral-Small-24B-Base",
+        "unsloth/Mistral-Small-24B-Base-2501",
         "mistralai/Mistral-Small-24B-Base-2501",
         "unsloth/Mistral-Small-24B-Base-2501-bnb-4bit",
     ),
     "unsloth/Mistral-Small-24B-Instruct-2501-unsloth-bnb-4bit" : (
-        "unsloth/Mistral-Small-24B-Instruct",
+        "unsloth/Mistral-Small-24B-Instruct-2501",
         "mistralai/Mistral-Small-24B-Instruct-2501",
         "unsloth/Mistral-Small-24B-Instruct-2501-bnb-4bit",
     ),