From c6328788cabd65476f825a6427a7ddddd596474c Mon Sep 17 00:00:00 2001
From: M Essam Hamed <github@messam.xyz>
Date: Tue, 19 Aug 2025 16:03:26 +0300
Subject: [PATCH] Resolve comments

---
 client/internal/updatemanager/manager.go | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/client/internal/updatemanager/manager.go b/client/internal/updatemanager/manager.go
index 4cc7ee73e..cb41f5370 100644
--- a/client/internal/updatemanager/manager.go
+++ b/client/internal/updatemanager/manager.go
@@ -61,15 +61,20 @@ func (u *UpdateManager) SetVersion(v string) {
 		log.Tracef("Auto-update version set to %s", v)
 		u.version = v
 		u.mutex.Unlock()
-		go u.Updated("N/A")
+		go u.Updated(unknownVersion)
 	} else {
 		u.mutex.Unlock()
 	}
 }
 
 func (u *UpdateManager) Stop() {
-	u.update.StopWatch()
 	u.cancel()
+	u.mutex.Lock()
+	defer u.mutex.Unlock()
+	if u.update != nil {
+		u.update.StopWatch()
+		u.update = nil
+	}
 	u.waitGroup.Wait()
 }
 
@@ -78,12 +83,7 @@ func (u *UpdateManager) Updated(latestVersion string) {
 	defer u.waitGroup.Done()
 	u.mutex.Lock()
 	defer u.mutex.Unlock()
-	select {
-	case <-u.ctx.Done():
-		return
-	default:
-	}
-	if latestVersion != "N/A" {
+	if latestVersion != unknownVersion {
 		u.latestVersion = latestVersion
 	}
 	ctx, cancel := context.WithDeadline(u.ctx, time.Now().Add(time.Minute))