Fixed buggy line of generated code