diff --git a/tunnel/freedom/freedom_test.go b/tunnel/freedom/freedom_test.go
index 7167fbe..041fb9a 100644
--- a/tunnel/freedom/freedom_test.go
+++ b/tunnel/freedom/freedom_test.go
@@ -72,7 +72,7 @@ func TestSocks(t *testing.T) {
 		noDelay:      true,
 	}
 	target, err := tunnel.NewAddressFromAddr("tcp", util.EchoAddr)
-
+	common.Must(err)
 	s, _ := socks5.NewClassicServer(socksAddr.String(), "127.0.0.1", "", "", 0, 0)
 	s.Handle = &socks5.DefaultHandle{}
 	go s.RunTCPServer()
diff --git a/tunnel/trojan/trojan_test.go b/tunnel/trojan/trojan_test.go
index cda1416..4fb0452 100644
--- a/tunnel/trojan/trojan_test.go
+++ b/tunnel/trojan/trojan_test.go
@@ -52,13 +52,15 @@ func TestTrojan(t *testing.T) {
 	c, err := NewClient(clientCtx, tcpClient)
 	common.Must(err)
 	s, err := NewServer(serverCtx, tcpServer)
-
+	common.Must(err)
 	conn1, err := c.DialConn(&tunnel.Address{
 		DomainName:  "example.com",
 		AddressType: tunnel.DomainName,
 	}, nil)
+	common.Must(err)
 	common.Must2(conn1.Write([]byte("87654321")))
 	conn2, err := s.AcceptConn(nil)
+	common.Must(err)
 	buf := [8]byte{}
 	conn2.Read(buf[:])
 	if !util.CheckConn(conn1, conn2) {
diff --git a/tunnel/websocket/server.go b/tunnel/websocket/server.go
index 7ca0a86..9482fcd 100644
--- a/tunnel/websocket/server.go
+++ b/tunnel/websocket/server.go
@@ -92,6 +92,9 @@ func (s *Server) AcceptConn(tunnel.Tunnel) (tunnel.Conn, error) {
 	url := "wss://" + s.hostname + s.path
 	origin := "https://" + s.hostname
 	wsConfig, err := websocket.NewConfig(url, origin)
+	if err != nil {
+		return nil, common.NewError("failed to create websocket config").Base(err)
+	}
 	var wsConn *websocket.Conn
 	ctx, cancel := context.WithCancel(s.ctx)