diff --git a/src/core/router.go b/src/core/router.go
index a7a69a8..ca805d7 100644
--- a/src/core/router.go
+++ b/src/core/router.go
@@ -78,7 +78,7 @@ func (rm *RouterManage) RouterSend(urlPath string, request Request, response Res
 	vc := reflect.New(rm.RegisteredData[ctr])
 
 	wsinit := vc.MethodByName("WebSocketInit")
-	if wsinit.IsValid() {
+	if wsinit.IsValid() || request.HEADER["Upgrade"] == "websocket" {
 		response.IsWebSocket = true
 		hand := vc.Interface().(interfaces.WebSocketInterface)
 		ws := GetWebSocket(ctr+"-"+fun, hand)