A configuration is disclosed to enable a UE to determine and/or prepare Rx beams prior to downlink reception in multi-TRP operation. The apparatus sends, to a first TRP, a report indicating a plurality of threshold scheduled offsets. Each threshold scheduled offset associated with a different number of TCI states. The apparatus receives, from the first TRP, DCI including a TCI and a time-offset. The TCI comprising at least one TCI state. Each TCI state indicating a Tx beam for a TRP of a set of TRPs. The set of TRPs including the first TRP. The apparatus determines at least one Rx beam based on at least one of the received TCI, the time-offset, or one or more of the plurality of threshold scheduled offsets. The apparatus receives, based on the received time-offset, a PDSCH from each TRP of the set of TRPs through the determined at least one Rx beam.