Fix incorrect keys in diagram.