Fixed the diagram reference