Various aspects of the present disclosure generally relate to wireless communication. In some aspects, a user equipment (UE) may determine, based at least in part on a determination that a scheduling offset is less than a beam switching latency threshold, a set of beams for communicating with a set of transmit-receive points (TRPs), that use single downlink control information for scheduling, based at least in part on at least one of a transmission configuration indicator codepoint mapping or a control resource set configuration. The UE may communicate with the set of TRPs using the set of beams based at least in part on determining the set of beams. Numerous other aspects are provided.