Fix for test failure