Fix to diagram