A method performed by a wireless device (510, 700, 1100) is disclosed. The wireless device obtains (601) a configuration for a sub-band channel quality indicator (CQI) granularity and a sub-band precoding matrix indicator (PMI) granularity for the wireless device. The wireless device determines (602) channel state information (CSI) feedback according to the configured sub-band CQI granularity and the sub-band PMI granularity. The sub-band PMI granularity corresponds to a first sub-band size and the sub-band CQI granularity corresponds to a second sub-band size. The first sub-band size is smaller than the second sub-band size. The wireless device transmits (603), to a network node (560, 1000), the determined CSI feedback.