I'm not sure if we should change this since it's not clear to me whether the unit of ``z`` should match that of ``rho`` in general. I'd prefer to punt on this and implement if we feel the need to enforce this.