A client device for associating phase tracking reference signal (PTRS) antenna ports to demodulation reference signal (DMRS) antenna ports, the client device being configured obtain configuration information related to a set of PTRS antenna ports, obtain control information indicating at least one of one or more codewords, a set of DMRS antenna ports assigned to the one or more codewords, or transmission configuration indication (TCI) information indicating one or more TCI states, and determine an association between the set of PTRS antenna ports and the set of DMRS antenna ports based on the configuration information related to the set of PTRS antenna ports and the control information.