A method in a wireless communication User Equipment (UE) includes receiving a Physical Downlink Control Channel (PDCCH) message assigning a data transmission on a Physical Downlink Shared CHannel (PDSCH), wherein the PDCCH message includes scheduling information with a first New Data Indicator (NDI) bit field for a first transport block and a second New Data Indicator (NDI) bit field for a second transport block. The UE determines one or more antenna ports for the data transmission based on the NDI bit fields of the transport blocks, before receiving the assigned data transmission on the PDSCH.