I think it might be useful to keep in the `highWaterMark` change to verify that this is working as intended. The old behavior of first would still pass this test case if `input` is smaller than the default `highWaterMark`.