A method performed by a User Equipment, UE, for communicating with a plurality of Transmission Reception Points, TRPs, including a first TRP and a second TRP, the method comprising: receiving, from a Base Station, BS, one or more Radio Resource Control, RRC, messages that configure the UE with a first Sounding Reference Signal, SRS, resource set associated with the first TRP, a second SRS resource set associated with the second TRP, and a round-robin-based mapping rule for a plurality of Physical Uplink Shared Channel, PUSCH, repetitions, the first SRS resource set and the second SRS resource set being configured for codebook-based transmissions; and transmitting the plurality of PUSCH repetitions based on the round-robin-based mapping rule at least by: transmitting a first set of PUSCH repetitions in the plurality of PUSCH repetitions to the first TRP according to the first SRS resource set and the round-robin-based mapping rule; and transmitting a second set of PUSCH repetitions of the plurality of PUSCH repetitions to the second TRP according to the second SRS resource set and the round-robin-based mapping rule, wherein: the first set of PUSCH repetitions are odd PUSCH repetitions of the plurality of PUSCH repetitions and the second set of PUSCH repetitions are even PUSCH repetitions of the plurality of PUSCH repetitions, the one or more RRC messages comprise a first Transmit Precoder Matrix Indication, TPMI, associated with the first SRS resource set and a second TPMI associated with the second SRS resource set, the first PUSCH repetition is mapped to the first TPMI and the second PUSCH repetition is mapped to the second TPMI according to the round-robin-based mapping rule, and the one or more RRC messages further comprise a first SRS Resource Indicator, SRI, corresponding to the first SRS resource set and a second SRI corresponding to the second SRS resource set. receiving, from a Base Station, BS, one or more Radio Resource Control, RRC, messages that configure the UE with a first Sounding Reference Signal, SRS, resource set associated with the first TRP, a second SRS resource set associated with the second TRP, and a round-robin-based mapping rule for a plurality of Physical Uplink Shared Channel, PUSCH, repetitions, the first SRS resource set and the second SRS resource set being configured for codebook-based transmissions; and transmitting the plurality of PUSCH repetitions based on the round-robin-based mapping rule at least by: transmitting a first set of PUSCH repetitions in the plurality of PUSCH repetitions to the first TRP according to the first SRS resource set and the round-robin-based mapping rule; and transmitting a second set of PUSCH repetitions of the plurality of PUSCH repetitions to the second TRP according to the second SRS resource set and the round-robin-based mapping rule, wherein: the first set of PUSCH repetitions are odd PUSCH repetitions of the plurality of PUSCH repetitions and the second set of PUSCH repetitions are even PUSCH repetitions of the plurality of PUSCH repetitions, the one or more RRC messages comprise a first Transmit Precoder Matrix Indication, TPMI, associated with the first SRS resource set and a second TPMI associated with the second SRS resource set, the first PUSCH repetition is mapped to the first TPMI and the second PUSCH repetition is mapped to the second TPMI according to the round-robin-based mapping rule, and the one or more RRC messages further comprise a first SRS Resource Indicator, SRI, corresponding to the first SRS resource set and a second SRI corresponding to the second SRS resource set. transmitting a first set of PUSCH repetitions in the plurality of PUSCH repetitions to the first TRP according to the first SRS resource set and the round-robin-based mapping rule; and transmitting a second set of PUSCH repetitions of the plurality of PUSCH repetitions to the second TRP according to the second SRS resource set and the round-robin-based mapping rule, wherein: the first set of PUSCH repetitions are odd PUSCH repetitions of the plurality of PUSCH repetitions and the second set of PUSCH repetitions are even PUSCH repetitions of the plurality of PUSCH repetitions, the one or more RRC messages comprise a first Transmit Precoder Matrix Indication, TPMI, associated with the first SRS resource set and a second TPMI associated with the second SRS resource set, the first PUSCH repetition is mapped to the first TPMI and the second PUSCH repetition is mapped to the second TPMI according to the round-robin-based mapping rule, and the one or more RRC messages further comprise a first SRS Resource Indicator, SRI, corresponding to the first SRS resource set and a second SRI corresponding to the second SRS resource set. the first set of PUSCH repetitions are odd PUSCH repetitions of the plurality of PUSCH repetitions and the second set of PUSCH repetitions are even PUSCH repetitions of the plurality of PUSCH repetitions, the one or more RRC messages comprise a first Transmit Precoder Matrix Indication, TPMI, associated with the first SRS resource set and a second TPMI associated with the second SRS resource set, the first PUSCH repetition is mapped to the first TPMI and the second PUSCH repetition is mapped to the second TPMI according to the round-robin-based mapping rule, and the one or more RRC messages further comprise a first SRS Resource Indicator, SRI, corresponding to the first SRS resource set and a second SRI corresponding to the second SRS resource set.