Fix test after 916ddc60df64518190d767fc48e032a193881ae2