Fix spacing in diagram