A solution for establishing a data traffic (e.g., PDU) session for a user equipment (UE) on a network (e.g., 5G cellular network) comprises receiving, by an access node (e.g., AMF), from a network repository (e.g., NRF), identification of a plurality of available session management nodes (e.g., SMFs), wherein the identification of the plurality of available session management nodes indicates whether each session management node supports a first traffic type or a second traffic type (e.g., IPv4 or (IPv6); determining, by the access node, a requested traffic type identified by the UE; determining a match between supported and requested traffic types; based at least on the match, selecting a session management node from the plurality of available session management nodes; and based at least on selecting the session management node, establishing the traffic session with the UE and the selected session management node.