diff --git a/fast_trace/fast_trace_test.go b/fast_trace/fast_trace_test.go
index b8763ae..285c3b8 100644
--- a/fast_trace/fast_trace_test.go
+++ b/fast_trace/fast_trace_test.go
@@ -1,14 +1,24 @@
 package fastTrace
 
 import (
+	"os"
+	"os/signal"
 	"testing"
 
 	"github.com/xgadget-lab/nexttrace/trace"
+	"github.com/xgadget-lab/nexttrace/wshandle"
 )
 
 // ICMP Use Too Many Time to Wait So we don't test it.
 func TestTCPTrace(t *testing.T) {
 	ft := FastTracer{}
+	// 建立 WebSocket 连接
+	w := wshandle.New()
+	w.Interrupt = make(chan os.Signal, 1)
+	signal.Notify(w.Interrupt, os.Interrupt)
+	defer func() {
+		w.Conn.Close()
+	}()
 	ft.TracerouteMethod = trace.TCPTrace
 	ft.testCM()
 	ft.testEDU()