Fix unit test error.