A method performed by a wireless device for Channel State Information, CSI, feedback, the method comprising: • receiving, from a base station, a configuration comprising: ∘ N Non-Zero Power, NZP, CSI Reference Signal, CSI-RS, resources, each of the N NZP CSI-RS resources associated with a different one of N Transmission Points, TRPs; or N sets of NZP CSI-RS resources, each of the N sets associated with a different one of the N TRPs; • receiving a request for CSI feedback based on the configuration; • selecting a preferred subset of the NZP CSI-RS resources comprised in the N NZP CSI-RS resources or the N sets of NZP CSI-RS resources, ∘ wherein, when the configuration comprises N NZP CSI-RS resources, the preferred subset of the NZP CSI-RS resources is a subset of the N NZP CSI-RS resources, and when the configuration comprises N sets of NZP CSI-RS resources, the preferred subset of the NZP CSI-RS resources is one or more of the N sets of NZP CSI-RS resources, ∘ wherein the preferred subset of the NZP CSI-RS resources has size M, where M<N; ∘ wherein selecting the preferred subset of the NZP CSI-RS resources corresponds to selecting M preferred TRPs out of the N TRPs, and ∘ wherein M is signaled by the base station to the wireless device; and • reporting, to the base station, CSI based on the selected preferred subset of the NZP CSI-RS resources, wherein the CSI comprises an identifier of each of the NZP CSI-RS resources in the preferred subset. • receiving, from a base station, a configuration comprising: ∘ N Non-Zero Power, NZP, CSI Reference Signal, CSI-RS, resources, each of the N NZP CSI-RS resources associated with a different one of N Transmission Points, TRPs; or N sets of NZP CSI-RS resources, each of the N sets associated with a different one of the N TRPs; ∘ N Non-Zero Power, NZP, CSI Reference Signal, CSI-RS, resources, each of the N NZP CSI-RS resources associated with a different one of N Transmission Points, TRPs; or N sets of NZP CSI-RS resources, each of the N sets associated with a different one of the N TRPs; • receiving a request for CSI feedback based on the configuration; • selecting a preferred subset of the NZP CSI-RS resources comprised in the N NZP CSI-RS resources or the N sets of NZP CSI-RS resources, ∘ wherein, when the configuration comprises N NZP CSI-RS resources, the preferred subset of the NZP CSI-RS resources is a subset of the N NZP CSI-RS resources, and when the configuration comprises N sets of NZP CSI-RS resources, the preferred subset of the NZP CSI-RS resources is one or more of the N sets of NZP CSI-RS resources, ∘ wherein the preferred subset of the NZP CSI-RS resources has size M, where M<N; ∘ wherein selecting the preferred subset of the NZP CSI-RS resources corresponds to selecting M preferred TRPs out of the N TRPs, and ∘ wherein M is signaled by the base station to the wireless device; and ∘ wherein, when the configuration comprises N NZP CSI-RS resources, the preferred subset of the NZP CSI-RS resources is a subset of the N NZP CSI-RS resources, and when the configuration comprises N sets of NZP CSI-RS resources, the preferred subset of the NZP CSI-RS resources is one or more of the N sets of NZP CSI-RS resources, ∘ wherein the preferred subset of the NZP CSI-RS resources has size M, where M<N; ∘ wherein selecting the preferred subset of the NZP CSI-RS resources corresponds to selecting M preferred TRPs out of the N TRPs, and ∘ wherein M is signaled by the base station to the wireless device; and • reporting, to the base station, CSI based on the selected preferred subset of the NZP CSI-RS resources, wherein the CSI comprises an identifier of each of the NZP CSI-RS resources in the preferred subset.