Changed expectation to be more readable