Handle imports during MethodLifting/RestoreMethods