Fix Unit test failure.