Fix for one unit test failure