diff --git a/doc/TRAINING_FLAGS.rst b/doc/TRAINING_FLAGS.rst
index deb3e65a..a0f2b48a 100644
--- a/doc/TRAINING_FLAGS.rst
+++ b/doc/TRAINING_FLAGS.rst
@@ -8,7 +8,7 @@ Below you can find the definition of all command-line flags supported by the tra
 Flags
 -----
 
-.. literalinclude:: ../training/coqui_stt_training/util/flags.py
+.. literalinclude:: ../training/coqui_stt_training/util/config.py
    :language: python
    :linenos:
    :lineno-match:
diff --git a/training/coqui_stt_training/util/config.py b/training/coqui_stt_training/util/config.py
index 0b072243..34794f1b 100755
--- a/training/coqui_stt_training/util/config.py
+++ b/training/coqui_stt_training/util/config.py
@@ -215,6 +215,7 @@ class _SttConfig(Coqpit):
             )
             sys.exit(1)
 
+    # sphinx-doc: training_ref_flags_start
     train_files: List[str] = field(
         default_factory=list,
         metadata=dict(
@@ -722,6 +723,7 @@ class _SttConfig(Coqpit):
             help="the number of trials to run during hyperparameter optimization."
         ),
     )
+    # sphinx-doc: training_ref_flags_end
 
 
 def initialize_globals_from_cli():