diff --git a/egs/tedlium2/ASR/conformer_ctc2/.train.py.swp b/egs/tedlium2/ASR/conformer_ctc2/.train.py.swp
index 1b7a548aa..5a08a9b0c 100644
Binary files a/egs/tedlium2/ASR/conformer_ctc2/.train.py.swp and b/egs/tedlium2/ASR/conformer_ctc2/.train.py.swp differ
diff --git a/egs/tedlium2/ASR/conformer_ctc2/train.py b/egs/tedlium2/ASR/conformer_ctc2/train.py
index d016fd8f1..1ee7bb4b7 100755
--- a/egs/tedlium2/ASR/conformer_ctc2/train.py
+++ b/egs/tedlium2/ASR/conformer_ctc2/train.py
@@ -1098,7 +1098,7 @@ def scan_pessimistic_batches_for_oom(
 
 def main():
     parser = get_parser()
-    LibriSpeechAsrDataModule.add_arguments(parser)
+    TedAsrDataModule.add_arguments(parser)
     args = parser.parse_args()
     args.exp_dir = Path(args.exp_dir)