Fix another test breakage.