Fix expectations for mock objects.