fix unit test mock