Systems and methods are disclosed herein for determining a power to be used for a set of antenna ports for a physical uplink shared channel transmission. In some embodiments, a User Equipment (UE) comprises processing circuitry configured to derive a power P to be used for uplink power control for a physical uplink shared channel transmission and determine a power to be used for a set of antenna ports based on the power P according to a rule that depends on whether the UE is utilizing codebook based transmission or non-codebook based transmission for the physical uplink shared channel transmission. The set of antenna ports is antenna ports on which the physical uplink shared channel transmission is transmitted with non-zero power.