diff --git a/egs/librispeech/ASR/conformer_ctc2/.train.py.swp b/egs/librispeech/ASR/conformer_ctc2/.train.py.swp
index c63e67c94..c1c319b8b 100644
Binary files a/egs/librispeech/ASR/conformer_ctc2/.train.py.swp and b/egs/librispeech/ASR/conformer_ctc2/.train.py.swp differ
diff --git a/egs/librispeech/ASR/conformer_ctc2/train.py b/egs/librispeech/ASR/conformer_ctc2/train.py
index 146d10017..d280ccdc0 100755
--- a/egs/librispeech/ASR/conformer_ctc2/train.py
+++ b/egs/librispeech/ASR/conformer_ctc2/train.py
@@ -56,6 +56,8 @@ from asr_datamodule import LibriSpeechAsrDataModule
 from conformer import Conformer
 from lhotse.dataset.sampling.base import CutSampler
 from lhotse.utils import fix_random_seed
+from lhotse.cut import Cut
+from lhotse.dataset.sampling.base import CutSampler
 import sys
 sys.path.append('/home/work/workspace/icefall/egs/tedlium2/ASR')
 from local.convert_transcript_words_to_bpe_ids import convert_texts_into_ids