diff --git a/egs/librispeech/ASR/.decode.sh.swp b/egs/librispeech/ASR/.decode.sh.swp
index 9cba9e201..c2d2cd59c 100644
Binary files a/egs/librispeech/ASR/.decode.sh.swp and b/egs/librispeech/ASR/.decode.sh.swp differ
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_lora.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_lora.py.swp
index 8b49a7ead..870d27e42 100644
Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_lora.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_lora.py.swp differ
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py
index 45f343565..52ab06ee0 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py
@@ -1147,7 +1147,7 @@ def train_one_epoch(
 
             if rank == 0: 
                 for i, lora in enumerate(lora_modules):
-                    lora.save_checkpoint(i, params.exp_dir)
+                    lora.save_checkpoint(i, params.batch_idx_train, params.exp_dir)
 
             '''
             remove_checkpoints(