Fixed error in diagram