Fix another failing test