From c2bae80b6689e3fd05b491ec6dcfb6628f1ae2ae Mon Sep 17 00:00:00 2001
From: Marco Dinis <marco.dinis@goteleport.com>
Date: Tue, 15 Oct 2024 09:22:23 +0100
Subject: [PATCH] fix context early cancelation

---
 lib/srv/discovery/status.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/srv/discovery/status.go b/lib/srv/discovery/status.go
index 4e268c30bf93..8d6960e557e6 100644
--- a/lib/srv/discovery/status.go
+++ b/lib/srv/discovery/status.go
@@ -413,9 +413,9 @@ func (s *Server) acquireSemaphoreForUserTask(userTaskName string) (releaseFn fun
 
 	// Once the lease parent context is canceled, the lease will be released.
 	ctxWithLease, cancel := context.WithCancel(lease)
-	defer cancel()
 
 	releaseFn = func() {
+		cancel()
 		lease.Stop()
 		if err := lease.Wait(); err != nil {
 			s.Log.WithError(err).WithField("semaphore", userTaskName).Warn("error cleaning up UserTask semaphore")