Diagram working again