fix for test failure