When a TRP performs a TRP beam sweep for a WD with multiple antenna arrangements and the WD is expected to perform an UL transmission (e.g., the WD has at least a threshold amount of data in its transmit buffer) and UL beam selection is based on DL beam selection, in one embodiment the WD takes the available output power of the different antenna arrangements in to account during RSRP calculations for the different candidate TRP TX beams such that the RSRP from an antenna arrangement that has lower available output power is weighted less compared to the RSRP measured from an antenna arrangement that has higher available output power.