diff --git a/egs/librispeech/ASR/pruned_transducer_stateless9/beam_search.py b/egs/librispeech/ASR/pruned_transducer_stateless9/beam_search.py
index c5ae0669e..605d15fa4 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless9/beam_search.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless9/beam_search.py
@@ -720,9 +720,9 @@ def greedy_search_batch(
 
             k = torch.sum(
                 (
-                    c[:, : context_size]
+                    c[:, :context_size]
                     == c[:, context_size : context_size + 1].expand_as(
-                      c[:, : context_size]
+                        c[:, :context_size]
                     )
                 ).int(),
                 dim=1,