diff --git a/Contract/OnDisconnectInterface.php b/Contract/OnDisconnectInterface.php
index e08f743..df530b1 100644
--- a/Contract/OnDisconnectInterface.php
+++ b/Contract/OnDisconnectInterface.php
@@ -9,10 +9,10 @@ interface OnDisconnectInterface
 
 
 	/**
-	 * @param Server $server
-	 * @param int $fd
+	 * @param Server|\Swoole\Coroutine\Http\Server $server
+	 * @param int                                  $fd
 	 * @return void
 	 */
-	public function onDisconnect(Server $server, int $fd): void;
+	public function onDisconnect(Server|\Swoole\Coroutine\Http\Server $server, int $fd): void;
 
 }
diff --git a/Contract/OnMessageInterface.php b/Contract/OnMessageInterface.php
index 5e5a98e..6e92e37 100644
--- a/Contract/OnMessageInterface.php
+++ b/Contract/OnMessageInterface.php
@@ -10,10 +10,10 @@ interface OnMessageInterface
 
 
 	/**
-	 * @param Server $server
-	 * @param Frame $frame
+	 * @param Server|\Swoole\Coroutine\Http\Server $server
+	 * @param Frame                                $frame
 	 * @return void
 	 */
-	public function onMessage(Server $server, Frame $frame): void;
+	public function onMessage(Server|\Swoole\Coroutine\Http\Server $server, Frame $frame): void;
 
 }