Fixed test breakage.