A method performed by a network node, the method comprising: - configuring at least two sounding reference signal, SRS, resource sets for a user equipment via a higher layer, wherein each SRS resource set comprises at least one SRS resource, and wherein the higher layer configuration of the SRS comprises a parameter usage and the value of the parameter is set to nonCodebook or Codebook; - scheduling at least one physical uplink shared channel, PUSCH, transmission for said user equipment via a downlink control information, DCI, wherein at least two SRS resources are indicated via a sounding reference signal resource indicator, SRI, field of the DCI, wherein each SRS resource is associated with a different SRS resource set; - mapping a given SRI bit field of the SRI field to the SRS resources indicated from associated SRS resource sets; - receiving from the user equipment a PUSCH that is transmitted using antenna ports associated with the indicated SRS resources; wherein, in the case when the usage is set to codebook, the DCI indicates as many transmitted precoding matrix index, TPMI, values as the number of SRS resources indicated via the SRI field of the DCI in which a first TPMI value indicates the precoder for the antenna ports corresponding to a first SRS resource and a second TPMI value indicates the precoder for the antenna ports corresponding to a second SRS resource and so on; and/or wherein, in the case when the usage is set to nonCodebook, each SRS resource in the SRS resource set configuration is configured with only one port, and wherein each SRS port has a one-to-one mapping with a demodulation reference signal, DMRS, port and thereby, a transmission layer associated with said PUSCH. - configuring at least two sounding reference signal, SRS, resource sets for a user equipment via a higher layer, wherein each SRS resource set comprises at least one SRS resource, and wherein the higher layer configuration of the SRS comprises a parameter usage and the value of the parameter is set to nonCodebook or Codebook; - scheduling at least one physical uplink shared channel, PUSCH, transmission for said user equipment via a downlink control information, DCI, wherein at least two SRS resources are indicated via a sounding reference signal resource indicator, SRI, field of the DCI, wherein each SRS resource is associated with a different SRS resource set; - mapping a given SRI bit field of the SRI field to the SRS resources indicated from associated SRS resource sets; - receiving from the user equipment a PUSCH that is transmitted using antenna ports associated with the indicated SRS resources; wherein, in the case when the usage is set to codebook, the DCI indicates as many transmitted precoding matrix index, TPMI, values as the number of SRS resources indicated via the SRI field of the DCI in which a first TPMI value indicates the precoder for the antenna ports corresponding to a first SRS resource and a second TPMI value indicates the precoder for the antenna ports corresponding to a second SRS resource and so on; and/or wherein, in the case when the usage is set to nonCodebook, each SRS resource in the SRS resource set configuration is configured with only one port, and wherein each SRS port has a one-to-one mapping with a demodulation reference signal, DMRS, port and thereby, a transmission layer associated with said PUSCH.