A method, apparatus and computer program is described, comprising: obtaining a first beam alignment dataset, wherein the first beam alignment dataset comprises measurement data for a first plurality of beam pair transmissions, wherein each beam pair transmission of the first plurality is between one of a plurality of communication beams of a first user device and one of a plurality of communication beams of a base station and wherein the first plurality of beam pair transmissions is a subset of all available beam pair transmissions between the first user device and the base station; and selecting a first beam pair combination for communications between the first user device and the base station, wherein the first beam pair combination comprises one of the plurality of beams of the first user device and one of the plurality of beams of the base station, wherein the means for selecting the first beam pair combination comprises a machine-learning model trained with a second beam alignment dataset.