From 2765bcfb89dfa052b1fe688967904d5aa5704879 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Zolt=C3=A1n=20Papp?= <zoltan.pmail@gmail.com>
Date: Mon, 17 Nov 2025 17:24:19 +0100
Subject: [PATCH] Restore in case of error

---
 client/ui/debug.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/client/ui/debug.go b/client/ui/debug.go
index 54fe49946..df1bb5f86 100644
--- a/client/ui/debug.go
+++ b/client/ui/debug.go
@@ -289,6 +289,8 @@ func (s *serviceClient) handleRunForDuration(
 		return
 	}
 
+	defer s.restoreServiceState(conn, initialState)
+
 	if err := s.collectDebugData(conn, initialState, params, progressUI); err != nil {
 		handleError(progressUI, err.Error())
 		return
@@ -299,8 +301,6 @@ func (s *serviceClient) handleRunForDuration(
 		return
 	}
 
-	s.restoreServiceState(conn, initialState)
-
 	progressUI.statusLabel.SetText("Bundle created successfully")
 }