diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py
index e709c8263c..d559232d1a 100644
--- a/unsloth/models/_utils.py
+++ b/unsloth/models/_utils.py
@@ -1084,7 +1084,7 @@ def unsloth_compile_transformers(
     sdap_bool_masks      : bool = True,
     sdpa_gqa_replace     : bool = True,
     sdpa_disable_compile : bool = True,
-    remove_causal_masks  : bool = True,
+    disable_causal_masks : bool = True,
     import_from_cache    : bool = False,
     compile_functions    : bool = True,
     fuse_lm_head         : bool = True,
@@ -1102,7 +1102,7 @@ def unsloth_compile_transformers(
             sdap_bool_masks      = sdap_bool_masks,
             sdpa_gqa_replace     = sdpa_gqa_replace,
             sdpa_disable_compile = sdpa_disable_compile,
-            remove_causal_masks  = remove_causal_masks,
+            disable_causal_masks = disable_causal_masks,
             import_from_cache    = import_from_cache,
             compile_functions    = compile_functions,
             fuse_lm_head         = fuse_lm_head,