1 files changed, 3 insertions, 1 deletions

diff --git a/client.go b/client.go
index 00455ab..d0716b1 100644
--- a/client.go
+++ b/client.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+        "context"
         "log"
         "net"
 
@@ -10,6 +11,7 @@ import (
 type ClientHandler struct {
         SocketListenOn  string
         WebsocketServer string
+        Context         context.Context
 }
 
 func (h *ClientHandler) ServiceConnection(proxyconn net.Conn) {
@@ -24,7 +26,7 @@ func (h *ClientHandler) ServiceConnection(proxyconn net.Conn) {
 
         log.Println("Connected to server")
 
-        serviceBoth(wsconn, proxyconn)
+        serviceBoth(wsconn, proxyconn, h.Context)
 }
 
 func (h *ClientHandler) Run() {