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 e6e589ab6..b8b50401d 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 0b0369754..843c27a80 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
@@ -259,7 +259,7 @@ class LoRAModule(nn.Module):
         nn.init.zeros_(self.lora_B)
         nn.init_normal_(self.lora_A)
 
-    def forward(self, x, layer_id=-1):
+    def forward(self, x, layer_idx=-1):
         x = x.transpose(0, 1)
         x = self.lora_A[layer_idx](x)
         x = self.lora_B[layer_idx](x)