Fix unit test check.