fix unit test fail