Fix unit test bug.