Fix unit test issue.