A User Equipment, UE, comprising: 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 being antenna ports on which the physical uplink shared channel transmission is transmitted with non-zero power, wherein the rule is such that, for a case of codebook based transmission, in order to determine the power to be used for the set of antenna ports based on the power P according to the rule, the processing circuitry is further configured to: derive a second power P' by scaling the power P with a ratio, R, wherein the ratio R is a first number antenna ports divided by a second number of antenna ports, wherein the first number of antenna ports is a number of antenna ports on which the physical uplink shared channel transmission is transmitted with non-zero power, and the second number of antenna ports is determined according to a multi-antenna transmission configuration as one of: (a) a third number of ports used in a precoder indicated by a New Radio base station, gNB, and (b) a fourth number of spatial layers indicated by the gNB; and equally divide the second power P' across the set of antenna ports on which the physical uplink shared channel transmission is transmitted with non-zero power. 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 being antenna ports on which the physical uplink shared channel transmission is transmitted with non-zero power, wherein the rule is such that, for a case of codebook based transmission, in order to determine the power to be used for the set of antenna ports based on the power P according to the rule, the processing circuitry is further configured to: derive a second power P' by scaling the power P with a ratio, R, wherein the ratio R is a first number antenna ports divided by a second number of antenna ports, wherein the first number of antenna ports is a number of antenna ports on which the physical uplink shared channel transmission is transmitted with non-zero power, and the second number of antenna ports is determined according to a multi-antenna transmission configuration as one of: (a) a third number of ports used in a precoder indicated by a New Radio base station, gNB, and (b) a fourth number of spatial layers indicated by the gNB; and equally divide the second power P' across the set of antenna ports on which the physical uplink shared channel transmission is transmitted with non-zero power. derive a second power P' by scaling the power P with a ratio, R, wherein the ratio R is a first number antenna ports divided by a second number of antenna ports, wherein the first number of antenna ports is a number of antenna ports on which the physical uplink shared channel transmission is transmitted with non-zero power, and the second number of antenna ports is determined according to a multi-antenna transmission configuration as one of: (a) a third number of ports used in a precoder indicated by a New Radio base station, gNB, and (b) a fourth number of spatial layers indicated by the gNB; and equally divide the second power P' across the set of antenna ports on which the physical uplink shared channel transmission is transmitted with non-zero power.