Adjust message and test expectation