diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.convolution.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.convolution.py.swp
index 007ad0f09..51828130c 100644
Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.convolution.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.convolution.py.swp differ
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_adapter.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_adapter.py.swp
new file mode 100644
index 000000000..f46c1f06d
Binary files /dev/null and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_adapter.py.swp differ
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/convolution.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/convolution.py
index 3b0a857bd..cb6dda2cf 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/convolution.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/convolution.py
@@ -77,5 +77,7 @@ class ConvolutionModule(nn.Module):
         x = self.activation(self.norm(x))
 
         x = self.pointwise_conv2(x)
+        x = x.transpose(1, 2)
+        x = self.layer_norm(x)
 
-        return x.transpose(1, 2)
+        return x