Changing expectations in order to get the tests pass