A wireless device transmits one or more maximum numbers of multiple-input multiple-output (MIMO) layers for reception of a physical downlink shared channel (PDSCH), of a cell, scheduled by downlink control information (DCI) from each transmit/receive point (TRP) of a plurality of TRPs of the cell. The wireless device receives one or more parameters indicating a first number of MIMO layers, for a first TRP of the plurality of TRPs of the cell, wherein the first number of MIMO layers is equal to or less than a first maximum number of MIMO layers, of the one or more maximum numbers of MIMO layers, corresponding to the first TRP. The wireless device receives one or more transport blocks with a second number of MIMO layers, wherein the second number is equal to or less than the first number.