diff --git a/nakadi-producer/src/test/java/org/zalando/nakadiproducer/snapshots/impl/SnapshotCreationServiceTest.java b/nakadi-producer/src/test/java/org/zalando/nakadiproducer/snapshots/impl/SnapshotCreationServiceTest.java
index a91f0484..c61a1840 100644
--- a/nakadi-producer/src/test/java/org/zalando/nakadiproducer/snapshots/impl/SnapshotCreationServiceTest.java
+++ b/nakadi-producer/src/test/java/org/zalando/nakadiproducer/snapshots/impl/SnapshotCreationServiceTest.java
@@ -73,7 +73,7 @@ public void testSnapshotSavedInBatches() {
         // when snapshot returns 5 item stream
         when(snapshotEventProvider.getSnapshot(PUBLISHER_EVENT_TYPE, null)).thenReturn(eventPayloads.subList(0, 3));
         when(snapshotEventProvider.getSnapshot(PUBLISHER_EVENT_TYPE, 2)).thenReturn(eventPayloads.subList(3, 5));
-        //when(snapshotEventProvider.getSnapshot(PUBLISHER_EVENT_TYPE, 5)).thenReturn(Collections.emptyList());
+        when(snapshotEventProvider.getSnapshot(PUBLISHER_EVENT_TYPE, 4)).thenReturn(Collections.emptyList());
 
         // create a snapshot
         eventTransmissionService.createSnapshotEvents(PUBLISHER_EVENT_TYPE);