fix unit test fail error