Fix unit test failure.