Certain aspects of the present disclosure provide techniques for applying uplink transmission configuration indicator (TCI) states with downlink reference signals to codebook based physical uplink shared channel (PUSCH) transmissions. An example method generally includes receiving, from a network entity, signaling of an uplink transmission configuration indicator (TCI) state with a target codebook based uplink transmission signal, determining if the TCI state has a source downlink reference signal (RS) and deciding how to process the codebook based uplink transmission based on the determination.