fix unit test error.