Fix breakage from Diagram type change.