Fix another test issue