Fix unit test case.