Fix legit test breakage