Fix test breakage.