From e5679050244bd5930f58b7f5d4606a89f71729bd Mon Sep 17 00:00:00 2001
From: Dave Johnston <davejohnston@improbable.io>
Date: Wed, 16 Dec 2020 13:58:11 +0000
Subject: [PATCH] fix: use correct code in cleanup

---
 websocket_transport.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/websocket_transport.go b/websocket_transport.go
index 7631fc8..19cf95f 100644
--- a/websocket_transport.go
+++ b/websocket_transport.go
@@ -168,7 +168,7 @@ func (t *WebsocketTransport) cleanup(code websocket.StatusCode) error {
 		t.queue = nil
 	}
 	if t.wsConn != nil {
-		err = t.wsConn.Close(websocket.StatusGoingAway, "Done")
+		err = t.wsConn.Close(code, "Done")
 		t.wsConn = nil
 	}
 	if t.closeFunc != nil {