Add appropriate return value for mocked function