Small fix in the generator.