1 files changed, 8 insertions, 1 deletions

diff --git a/client.go b/client.go
index d0716b1..d9a0f25 100644
--- a/client.go
+++ b/client.go
@@ -26,7 +26,14 @@ func (h *ClientHandler) ServiceConnection(proxyconn net.Conn) {
 
         log.Println("Connected to server")
 
-        serviceBoth(wsconn, proxyconn, h.Context)
+        errc := make(chan error)
+        ws := &WebsocketReadWriter{wsconn}
+
+        go serviceBoth(proxyconn, ws, errc)
+        go serviceBoth(ws, proxyconn, errc)
+
+        <-errc
+        log.Println("Closing client connection")
 }
 
 func (h *ClientHandler) Run() {