fix generator.