diff --git a/unsloth/models/mistral.py b/unsloth/models/mistral.py
index 2941fb32fb..91a2dd7e35 100644
--- a/unsloth/models/mistral.py
+++ b/unsloth/models/mistral.py
@@ -375,6 +375,9 @@ class FastMistralModel(FastLlamaModel):
         
         # Log Unsloth version for future fastpaths for inference
         model.config.update({"unsloth_version" : __version__})
+
+        # Add save modules
+        patch_saving_functions(model)
         
         return model, tokenizer
     pass
diff --git a/unsloth/save.py b/unsloth/save.py
index baa8f3f5ca..4cda8b9651 100644
--- a/unsloth/save.py
+++ b/unsloth/save.py
@@ -278,7 +278,6 @@ def unsloth_save_model(
         not hasattr(internal_model.model, "layers")
     ):
         # Do general saving
-        print(type(model))
         # Edit save_pretrained_settings
         # [TODO] _create_repo has errors due to **kwargs getting accepted
         for deletion in \