Adjust expectation in a unit test