diff --git a/egs/tedlium2/ASR/conformer_ctc2/.train.py.swp b/egs/tedlium2/ASR/conformer_ctc2/.train.py.swp
index 510ed8ec5..7e1995bf3 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 edb4dfdf7..6c6149955 100755
--- a/egs/tedlium2/ASR/conformer_ctc2/train.py
+++ b/egs/tedlium2/ASR/conformer_ctc2/train.py
@@ -58,7 +58,7 @@ import optim
 import torch
 import torch.multiprocessing as mp
 import torch.nn as nn
-from asr_datamodule import LibriSpeechAsrDataModule
+from asr_datamodule import LibriSpeechAsrDataModule, TedAsrDataModule
 from conformer import Conformer
 from lhotse.cut import Cut
 from lhotse.dataset.sampling.base import CutSampler