diff --git a/tunnel/trojan/trojan_test.go b/tunnel/trojan/trojan_test.go
index 4decaf4..b5c232e 100644
--- a/tunnel/trojan/trojan_test.go
+++ b/tunnel/trojan/trojan_test.go
@@ -75,9 +75,11 @@ func TestTrojan(t *testing.T) {
 		redir:      redirector.NewRedirector(ctx),
 	}
 	go s.acceptLoop()
+	ctx, cancel = context.WithCancel(ctx)
 	c := &Client{
 		underlay: tcpClient,
 		ctx:      ctx,
+		cancel:   cancel,
 		user:     &MockUser{},
 	}