diff --git a/src/runtime/communication.rs b/src/runtime/communication.rs
index 83e5ede6e7378bc849393b85fe82ca92a76cf8cf..4b55cea8d8da2020cd3d36719bcfc5694ae9df0a 100644
--- a/src/runtime/communication.rs
+++ b/src/runtime/communication.rs
@@ -114,15 +114,8 @@ impl Connector {
         }
     }
     // #[cfg(ffi_socket_api)]
-    pub(crate) fn get_mut_udp_sock(&mut self, index: usize) -> Option<&mut UdpSocket> {
-        let sock = &mut self
-            .get_comm_mut()?
-            .endpoint_manager
-            .udp_endpoint_store
-            .endpoint_exts
-            .get_mut(index)?
-            .sock;
-        Some(sock)
+    pub(crate) fn get_mut_udp_ee(&mut self, index: usize) -> Option<&mut UdpEndpointExt> {
+        self.get_comm_mut()?.endpoint_manager.udp_endpoint_store.endpoint_exts.get_mut(index)
     }
     pub fn gotten(&mut self, port: PortId) -> Result<&Payload, GottenError> {
         use GottenError as Ge;