Fix another test