Fixing a bug in the generator.