A configuration of a plurality of transmission configuration indicator go) states for a serving cell is received for physical downlink shared channel (PDSCH) transmissions, wherein each of the plurality of TCI states contains parameters for configuring a quasi-co-location relationship between downlink reference signals and demodulation reference signal ports of the PDSCH transmissions. A physical downlink control channel (PDCCH) is received, which includes downlink control information for scheduling a PDSCH. A determination is made as to whether an offset between the reception of the PDCCH and a reception of the PDSCH is less than a predetermined threshold, where in response to the determination that the offset is less than the predetermined threshold, a configured pair of TCI states is selected from the received configuration of the plurality of TCI states. The PDSCH is decoded based on the selected configured pair of TCI states.