From cd19f4d910cc115709e8c11031b48b54faa9d433 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Zolt=C3=A1n=20Papp?= <zoltan.pmail@gmail.com>
Date: Mon, 13 Oct 2025 20:47:52 +0200
Subject: [PATCH] Code cleaning in updateState

---
 client/internal/updatemanager/manager.go | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/client/internal/updatemanager/manager.go b/client/internal/updatemanager/manager.go
index 2b4562430..a97fa61db 100644
--- a/client/internal/updatemanager/manager.go
+++ b/client/internal/updatemanager/manager.go
@@ -262,15 +262,18 @@ func (u *UpdateManager) handleUpdate(ctx context.Context) {
 }
 
 func (u *UpdateManager) updateStateManager(ctx context.Context) {
-	u.stateManager.RegisterState(&UpdateState{})
-	if err := u.stateManager.LoadState(&UpdateState{}); err != nil {
+	var stateType *UpdateState
+
+	u.stateManager.RegisterState(stateType)
+	if err := u.stateManager.LoadState(stateType); err != nil {
 		log.Errorf("failed to load state: %v", err)
 		return
 	}
-	state := u.stateManager.GetState(&UpdateState{})
+	state := u.stateManager.GetState(stateType)
 	if state == nil {
 		return
 	}
+
 	updateState, ok := state.(*UpdateState)
 	if !ok {
 		log.Errorf("failed to cast state to UpdateState")