Fix the mocking test error