Various solutions for improvement of scheduling multi-cell PUSCH/PDSCH transmission with a single DCI are described. An apparatus may receive a DCI scheduling cells from a network node. The DCI includes at least one of a first DCI, a second DCI, and a third DCI. The apparatus may determine a DCI size budget of the DCI counted in one cell. The apparatus may perform a DCI decoding according to the DCI size budget and perform a PDSCH reception or a PUSCH transmission with at least one cell based on the DCI. The first DCI includes a common bit field, a first specific bit field and a plurality of designated bit fields corresponding to the cells, the second DCI includes the common bit field, a second specific bit field and a first part of the designated bit fields, and the third DCI includes a second part of the designated bit fields.