Fix generation error