Fix another test failure