From 9cf1ddabb05193904f3f60474004047e9fb22f79 Mon Sep 17 00:00:00 2001
From: Simon Rozet <me@simonrozet.com>
Date: Fri, 10 Feb 2023 18:46:04 +0100
Subject: [PATCH] fix cancellation bug

---
 main.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/main.go b/main.go
index 9fcc2c8..8c6d129 100644
--- a/main.go
+++ b/main.go
@@ -249,6 +249,7 @@ func tsproxy(ctx context.Context) error {
 			if err := srv.Close(); err != nil {
 				log.Error("server shutdown", err)
 			}
+			cancel()
 		})
 		g.Add(func() error {
 			_, err := tsWaitStatusReady(ctx, log, lc)
@@ -264,6 +265,7 @@ func tsproxy(ctx context.Context) error {
 			if err := srv.Close(); err != nil {
 				log.Error("TLS server shutdown", err)
 			}
+			cancel()
 		})
 	}
 
-- 
2.51.2