diff --git a/tests/e2e/test_full_workflows.py b/tests/e2e/test_full_workflows.py
index 867eb69..f2b5bee 100644
--- a/tests/e2e/test_full_workflows.py
+++ b/tests/e2e/test_full_workflows.py
@@ -80,7 +80,7 @@ async def test_large_batch_streaming(
     batches = [l for l in lines if l["type"] == "batch"]
     complete = lines[-1]
 
-    assert len(batches) == 4
+    assert len(batches) == 5  # first batch is smaller (FIRST_BATCH_SIZE=5), then 4x50
     assert complete["total"] == 200