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 8a737e284..603fd4ff4 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 f20ca6260..33010e3ff 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
@@ -95,9 +95,7 @@ class TransformerEncoderAdapter(TransformerEncoder):
                 x, (z, lr) = layer(
                     x, self_attn_padding_mask=padding_mask, need_weights=False,
                 )
-                print(x.size())
-                x_ = self.adapters(x)
-                print(x_.size())
+                x = self.adapters(x, layer_id=i)
                 exit()
 
                 if i >= min_layer: