Use a mock for test