Corrected code generation when doing the second run with an 'infinite' loop.