Fixing type in diagram