diff --git a/internal/command/vnc/vm.go b/internal/command/vnc/vm.go index b41b275..990a5ba 100644 --- a/internal/command/vnc/vm.go +++ b/internal/command/vnc/vm.go @@ -66,6 +66,8 @@ func runVNCVM(cmd *cobra.Command, args []string) (err error) { } go func() { + defer conn.Close() + wsConn, err := client.VMs().PortForward(cmd.Context(), name, vncPort, wait) if err != nil { fmt.Printf("failed to forward port: %v\n", err) diff --git a/internal/command/vnc/worker.go b/internal/command/vnc/worker.go index 3d67658..57bb885 100644 --- a/internal/command/vnc/worker.go +++ b/internal/command/vnc/worker.go @@ -56,6 +56,8 @@ func runVNCWorker(cmd *cobra.Command, args []string) (err error) { } go func() { + defer conn.Close() + wsConn, err := client.Workers().PortForward(cmd.Context(), name, vncPort) if err != nil { fmt.Printf("failed to forward port: %v\n", err)