Fix another test I broke