diff --git a/core/Websocket.php b/core/Websocket.php
index 9a336d9..2753e03 100644
--- a/core/Websocket.php
+++ b/core/Websocket.php
@@ -21,8 +21,20 @@ abstract class Websocket implements OnMessageInterface, OnDisconnectInterface
 	public Transport $transport;
 
 
+	/**
+	 * @var Server
+	 */
 	private Server $server;
 
+
+	/**
+	 * @return Server
+	 */
+	public function getServer(): Server
+	{
+		return $this->server;
+	}
+
 	/**
 	 * @param string $host
 	 * @param int    $port
diff --git a/test.php b/test.php
index b51a6bf..d08122e 100644
--- a/test.php
+++ b/test.php
@@ -1,12 +1,10 @@
 <?php
 
 
-use Coroutine\Server\Transport;
 use Coroutine\Server\Websocket;
 use Kiri\Router\Constrict\ConstrictRequest;
 use Swoole\Coroutine\Http\Client;
 use Swoole\Coroutine\Http\Server;
-use Swoole\Http\Request;
 use Swoole\Http\Response;
 use Swoole\WebSocket\Frame;
 use function Co\run;