In one example, an indication that a user equipment participating in a Packet Data Network (PDN) session hosted by a Serving Gateway (SGW) and a PDN Gateway (PGW) is transitioning from a first Mobility Management Entity (MME) to a second MME is obtained. An indication that the SGW is co-located with the PGW and an identification of the SGW are obtained. Based on the indication that the SGW is co-located with the PGW and the identification of the SGW, it is determined that the SGW is reachable from the second MME. In response to determining that the SGW is reachable from the second MME, the SGW is selected to host the PDN session after the user equipment transitions from the first MME to the second MME.