diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/joiner.py b/egs/librispeech/ASR/pruned_transducer_stateless2/joiner.py
index 973a89bfe..d76a913a5 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless2/joiner.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless2/joiner.py
@@ -44,4 +44,4 @@ class Joiner(nn.Module):
 
         logit = self.output_linear(torch.tanh(logit))
 
-        return output
+        return logit