diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp
index b6120064c..8a2248644 100644
Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp differ
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode_new.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode_new.py.swp
index 044b279d3..c5b670539 100644
Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode_new.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode_new.py.swp differ
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode_new.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode_new.py
index 671edd23c..27ef63e5e 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode_new.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode_new.py
@@ -59,7 +59,8 @@ from beam_search import (
     greedy_search_batch,
     modified_beam_search,
 )
-from train import add_model_arguments, add_rep_arguments, get_params, get_transducer_model
+#from train import add_model_arguments, add_rep_arguments, get_params, get_transducer_model
+from prompt_tuning import add_model_arguments, add_rep_arguments, get_params, get_transducer_model
 
 from icefall.checkpoint import (
     average_checkpoints,