A wireless device receives a first downlink control information (DCI) indicating a transmission configuration indicator (TCI) codepoint that indicates a first TCI state and a second TCI state. The wireless device receives a second DCI comprising a sounding reference signal (SRS) resource set field indicating to apply one or both of the first TCI state and the second TCI state to a physical uplink shared channel (PUSCH) transmission. A value of the SRS resource set field is one of: a first value indicating to apply the first TCI state, a second value indicating to apply the second TCI state, and a third value indicating to apply the first TCI state and the second TCI state. The wireless device transmits repetitions of the PUSCH transmission based on the value of the SRS resource set field.