Fix a mock test