Diagram: Fixed the generator