diff --git a/src/epoll.c b/src/epoll.c
index 9a44b65..b788447 100644
--- a/src/epoll.c
+++ b/src/epoll.c
@@ -41,7 +41,7 @@ static int _ep_ctl_add(ep_port_t* port_info,
 static int _ep_ctl_mod(ep_port_t* port_info,
                        uintptr_t socket,
                        struct epoll_event* ev) {
-  ep_sock_t* sock_info = ep_sock_find_in_tree(&port_info->sock_tree, socket);
+  ep_sock_t* sock_info = ep_port_find_socket(port_info, socket);
   if (sock_info == NULL)
     return -1;
 
@@ -52,7 +52,7 @@ static int _ep_ctl_mod(ep_port_t* port_info,
 }
 
 static int _ep_ctl_del(ep_port_t* port_info, uintptr_t socket) {
-  ep_sock_t* sock_info = ep_sock_find_in_tree(&port_info->sock_tree, socket);
+  ep_sock_t* sock_info = ep_port_find_socket(port_info, socket);
   if (sock_info == NULL)
     return -1;
 
diff --git a/src/port.c b/src/port.c
index b7ec76e..e81ff1c 100644
--- a/src/port.c
+++ b/src/port.c
@@ -114,6 +114,10 @@ int ep_port_del_socket(ep_port_t* port_info, tree_node_t* tree_node) {
   return tree_del(&port_info->sock_tree, tree_node);
 }
 
+ep_sock_t* ep_port_find_socket(ep_port_t* port_info, SOCKET socket) {
+  return ep_sock_find_in_tree(&port_info->sock_tree, socket);
+}
+
 static poll_group_allocator_t* _ep_port_get_poll_group_allocator(
     ep_port_t* port_info,
     size_t protocol_id,
diff --git a/src/port.h b/src/port.h
index 60b6d4b..c208fea 100644
--- a/src/port.h
+++ b/src/port.h
@@ -44,6 +44,8 @@ EPOLL_INTERNAL int ep_port_add_socket(ep_port_t* port_info,
                                       SOCKET socket);
 EPOLL_INTERNAL int ep_port_del_socket(ep_port_t* port_info,
                                       tree_node_t* tree_node);
+EPOLL_INTERNAL ep_sock_t* ep_port_find_socket(ep_port_t* port_info,
+                                              SOCKET socket);
 
 EPOLL_INTERNAL void ep_port_request_socket_update(ep_port_t* port_info,
                                                   ep_sock_t* sock_info);