Revert one change

mafredri · mafredri · commit 2fa36d2dd05f · 2023-02-14T13:48:18.000Z
diff --git a/codersdk/workspaceagents.go b/codersdk/workspaceagents.go
@@ -174,9 +174,7 @@ func (c *Client) DialWorkspaceAgent(ctx context.Context, agentID uuid.UUID, opti
 				options.Logger.Debug(ctx, "failed to dial", slog.Error(err))
 				continue
 			}
-			ctx, wsNetConn := websocketNetConn(ctx, ws, websocket.MessageBinary)
-			defer wsNetConn.Close()
-			sendNode, errChan := tailnet.ServeCoordinator(wsNetConn, func(node []*tailnet.Node) error {
+			sendNode, errChan := tailnet.ServeCoordinator(websocket.NetConn(ctx, ws, websocket.MessageBinary), func(node []*tailnet.Node) error {
 				return conn.UpdateNodes(node)
 			})
 			conn.SetNodeCallback(sendNode)
