diff --git a/src/cnlpt/_cli/train.py b/src/cnlpt/_cli/train.py
index 44dc1544..6d2f7738 100644
--- a/src/cnlpt/_cli/train.py
+++ b/src/cnlpt/_cli/train.py
@@ -337,11 +337,18 @@ def transformers_arg_option(field_name: str, *args, **kwargs):
 FreezeEncoderArg = Annotated[float, training_arg_option("freeze_encoder")]
 BiasFitArg = Annotated[bool, training_arg_option("bias_fit")]
 EvalsPerEpochArg = Annotated[int, training_arg_option("evals_per_epoch")]
-RichDisplayArg = Annotated[bool, training_arg_option("rich_display")]
+RichDisplayArg = Annotated[
+    bool, training_arg_option("rich_display", "--rich_display/--no_rich_display")
+]
 LoggingStrategyArg = Annotated[
     IntervalStrategy, training_arg_option("logging_strategy")
 ]
-LoggingFirstStepArg = Annotated[bool, training_arg_option("logging_first_step")]
+LoggingFirstStepArg = Annotated[
+    bool,
+    training_arg_option(
+        "logging_first_step", "--logging_first_step/--no_logging_first_step"
+    ),
+]
 CacheDirArg = Annotated[Union[str, None], training_arg_option("cache_dir")]
 MetricForBestModelArg = Annotated[str, training_arg_option("metric_for_best_model")]