diff --git a/src/OnReceiveInterface.php b/src/OnReceiveInterface.php
index 3b44877..09dbae0 100644
--- a/src/OnReceiveInterface.php
+++ b/src/OnReceiveInterface.php
@@ -13,11 +13,12 @@
 
 use Swoole\Coroutine\Server\Connection;
 use Swoole\Server as SwooleServer;
+use Swow\Socket;
 
 interface OnReceiveInterface
 {
     /**
-     * @param Connection|SwooleServer $server
+     * @param Connection|Socket|SwooleServer $server
      */
     public function onReceive($server, int $fd, int $reactorId, string $data): void;
 }