The invention refers to a method performed by a radio network node (200), wherein the radio access network comprises a central unit, CU (220), and a distributed unit, DU (210), comprising obtaining (502) one or more criteria for determining a preferable location for a UE context controller in the radio access network node, the UE context controller managing a connection between a UE (100) and the radio access network node (200); and determining (504), based on the one or more criteria, whether the UE context controller shall be located in the CU (220) and/or in the DU (210). The invention further related to a corresponding radio network node (200).