Fix expectation for streaming response