A session management function (SMF) receives a message from a network exposure function (NEF). The message comprises downlink data and a release assistance indicator (RAI). The RAI indicates that transmission of uplink data by a wireless device is expected subsequent to transmission of the downlink data. The RAI indicates a release of a connection associated with the wireless device after the transmission of uplink data. The SMF receives an indication that the uplink data is transmitted. The SMF upon receiving the indication, sends a release message to an access and mobility management function (AMF). The release message indicates release of a non-access stratum message associated with the wireless device. The release message comprises the release assistance indicator.