Leonardo set for Milan return