diff --git a/mace/tools/arg_parser.py b/mace/tools/arg_parser.py
index 6d8be783..cb4f8ac5 100644
--- a/mace/tools/arg_parser.py
+++ b/mace/tools/arg_parser.py
@@ -361,7 +361,7 @@ def build_default_arg_parser() -> argparse.ArgumentParser:
         default=True,
     )
     parser.add_argument(
-        "--foundation_model_head",
+        "--foundation_head",
         help="Name of the head to use for fine-tuning",
         type=str,
         default=None,