diff --git a/examples/src/main/java/kafka/examples/ExactlyOnceMessageProcessor.java b/examples/src/main/java/kafka/examples/ExactlyOnceMessageProcessor.java
index 175385b..d424952 100644
--- a/examples/src/main/java/kafka/examples/ExactlyOnceMessageProcessor.java
+++ b/examples/src/main/java/kafka/examples/ExactlyOnceMessageProcessor.java
@@ -216,11 +216,7 @@
         }
         return consumer.assignment().stream().mapToLong(partition -> {
             long currentPosition = consumer.position(partition);
-            if (fullEndOffsets.containsKey(partition)) {
-                return fullEndOffsets.get(partition) - currentPosition;
-            } else {
-                return 0;
-            }
+            return fullEndOffsets.getOrDefault(partition, currentPosition) - currentPosition;
         }).sum();
     }