Make sure to insert wrapGenerator.mark calls at the Program level.