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 4f02a4272..8a737e284 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 94d8bc081..f20ca6260 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
@@ -154,8 +154,8 @@ class ResidualAdapterModule(nn.Module):
         self.adapter_layers = nn.ModuleList(
             [build_adapter(embedding_dim, proj_dim) for _ in range(layer_num)]
         )
-
-    def forward(x):#, layer_id=0):#, layer_id=-1):
+    
+    def forward(self, x, layer_id=-1):
         layer_id = 0
         print(x.size())
         x = x.transpose(0, 1)