Fix test compile