fix unit test failed