Fixed failing mocked test