Fix for the unit test failure.