diff --git a/egs/tedlium3/ASR/.lora.sh.swp b/egs/tedlium3/ASR/.lora.sh.swp
index 688dbd961..2bab2d475 100644
Binary files a/egs/tedlium3/ASR/.lora.sh.swp and b/egs/tedlium3/ASR/.lora.sh.swp differ
diff --git a/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/.train_tta.py.swp b/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/.train_tta.py.swp
index 1fa2e8fbc..2e6abe826 100644
Binary files a/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/.train_tta.py.swp and b/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/.train_tta.py.swp differ
diff --git a/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/train_tta.py b/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/train_tta.py
index 0468e016e..2050275ee 100755
--- a/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/train_tta.py
+++ b/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/train_tta.py
@@ -1659,7 +1659,7 @@ def scan_pessimistic_batches_for_oom(
 
 def main():
     parser = get_parser()
-    LibriSpeechAsrDataModule.add_arguments(parser)
+    TedliumAsrDataModule.add_arguments(parser)
     args = parser.parse_args()
     if args.wandb: args.exp_dir = args.exp_dir + str(random.randint(0,400))
     args.exp_dir = Path(args.exp_dir)