diff --git a/src/companion/forge/train.py b/src/companion/forge/train.py
index 86f604e..7682e19 100644
--- a/src/companion/forge/train.py
+++ b/src/companion/forge/train.py
@@ -211,7 +211,9 @@ def main():
         "--data", type=Path, required=True, help="Path to training data JSONL"
     )
     parser.add_argument(
+        "--output-dir",
         "--output",
+        dest="output",
         type=Path,
         default=Path("~/.companion/training"),
         help="Output directory",