diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.convolution.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.convolution.py.swp
index 8dca584b3..ee78edf41 100644
Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.convolution.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.convolution.py.swp differ
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp
index 4213cb476..4b53c9534 100644
Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp differ
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py
index d21bde89f..81cacba88 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py
@@ -32,6 +32,8 @@ from fairseq.modules import (
 from fairseq.utils import index_put
 from utils import pad_to_multiple
 
+from convolution import ConvolutionModule
+
 logger = logging.getLogger().setLevel(logging.INFO)
 
 class TransformerEncoderAdapter(TransformerEncoder):