fix unit test failed.