diff --git a/System/Snowflake.php b/System/Snowflake.php
index fe93510e..8847c8b0 100644
--- a/System/Snowflake.php
+++ b/System/Snowflake.php
@@ -197,12 +197,16 @@ class Snowflake
 
 
 	/**
-	 * @return Server|\Swoole\Http\Server
+	 * @return Server|null
 	 * @throws
 	 */
 	public static function getWebSocket()
 	{
-		return static::app()->get('server')->getServer();
+		$server = static::app()->get('server')->getServer();
+		if (!($server instanceof Server)) {
+			return null;
+		}
+		return $server;
 	}