Fixed an issue in the generation