Amendment to the generated code