Fix a bug in the generator