diff --git a/pulsar-broker/src/test/java/org/apache/pulsar/client/converter/CustomBatchConverter.java b/pulsar-broker/src/test/java/org/apache/pulsar/client/converter/CustomBatchConverter.java
index 080992c22cc667..2a00ad88f768db 100644
--- a/pulsar-broker/src/test/java/org/apache/pulsar/client/converter/CustomBatchConverter.java
+++ b/pulsar-broker/src/test/java/org/apache/pulsar/client/converter/CustomBatchConverter.java
@@ -48,7 +48,7 @@ public <T> Iterable<Message<T>> convert(EntryContext context, MessagePayload pay
         bufList.add(buf);
         final CustomBatchFormat.StringIterable strings = CustomBatchFormat.deserialize(buf);
         final Iterator<String> stringIterator = strings.iterator();
-        final int numMessages = context.getNumMessages();
+        final int numMessages = strings.size();
 
         return () -> new Iterator<Message<T>>() {
             int index = 0;