diff --git a/training/mochi-1/args.py b/training/mochi-1/args.py
index 4f1cc7e..51671c0 100644
--- a/training/mochi-1/args.py
+++ b/training/mochi-1/args.py
@@ -42,7 +42,7 @@ def _get_model_args(parser: argparse.ArgumentParser) -> None:
     parser.add_argument(
         "--compile_dit",
         action="store_true",
-        help="If we should cast DiT params to a lower precision.",
+        help="If we should compile the DiT.",
     )