diff --git a/egs/librispeech/ASR/conv_emformer_transducer_stateless/emformer.py b/egs/librispeech/ASR/conv_emformer_transducer_stateless/emformer.py
index c425fe67c..fa908ffcd 100644
--- a/egs/librispeech/ASR/conv_emformer_transducer_stateless/emformer.py
+++ b/egs/librispeech/ASR/conv_emformer_transducer_stateless/emformer.py
@@ -1252,8 +1252,8 @@ class EmformerEncoder(nn.Module):
     ):
         super().__init__()
 
-        assert int(math.log(chunk_length, 2)) == math.log(
-            chunk_length, 2
+        assert (
+            chunk_length - 1 & chunk_length == 0
         ), "chunk_length should be a power of 2."
 
         self.use_memory = memory_size > 0
diff --git a/egs/librispeech/ASR/conv_emformer_transducer_stateless2/emformer.py b/egs/librispeech/ASR/conv_emformer_transducer_stateless2/emformer.py
index c1175d395..18d2047bd 100644
--- a/egs/librispeech/ASR/conv_emformer_transducer_stateless2/emformer.py
+++ b/egs/librispeech/ASR/conv_emformer_transducer_stateless2/emformer.py
@@ -1188,8 +1188,8 @@ class EmformerEncoder(nn.Module):
     ):
         super().__init__()
 
-        assert int(math.log(chunk_length, 2)) == math.log(
-            chunk_length, 2
+        assert (
+            chunk_length - 1 & chunk_length == 0
         ), "chunk_length should be a power of 2."
 
         self.use_memory = memory_size > 0