diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.beam_search.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.beam_search.py.swp
index 18652a587..5620e87d4 100644
Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.beam_search.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.beam_search.py.swp differ
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.model.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.model.py.swp
index 5ec29cfae..8b1723b4d 100644
Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.model.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.model.py.swp differ
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py
index d20bd78ff..bef12f8fd 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py
@@ -205,7 +205,7 @@ class Transducer(nn.Module):
         y: k2.RaggedTensor,
         sp,
     ):
-        from beam_search import greedy_search_batch, greedy_search_batch_target_input
+        from beam_search import greedy_search_batch
 
         encoder_out, x_lens = self.encoder(x, x_lens)