diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py
index f443b91b27..505a012bb8 100644
--- a/unsloth/models/llama.py
+++ b/unsloth/models/llama.py
@@ -1399,7 +1399,7 @@ class FastLlamaModel:
                     (down_proj.base_layer if hasattr(down_proj, "base_layer") else down_proj).bias is None:
 
                     # https://stackoverflow.com/questions/50599045/python-replacing-a-function-within-a-class-of-a-module
-                    # layer.mlp.forward = types.MethodType(apply_lora_mlp, layer.mlp)
+                    layer.mlp.forward = types.MethodType(apply_lora_mlp, layer.mlp)
                     n_mlp += 1
                 else:
                     logger.warning_once(