Attempt to fix test failure