Attempted fix for test failure