Add wait for finsih in LegacyIdTransformation