Fix error in diagram