Fixing unit test failure.