From 1b97738e9e24b53ec92a8172749bf6830222a08b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?S=C3=B6ren=20Schneider?=
 <56670304+soerenschneider@users.noreply.github.com>
Date: Wed, 19 Jul 2023 22:57:26 +0100
Subject: [PATCH] fix: respect streak

---
 internal/agent/state/ok.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/internal/agent/state/ok.go b/internal/agent/state/ok.go
index e2e75e9..f1c1650 100644
--- a/internal/agent/state/ok.go
+++ b/internal/agent/state/ok.go
@@ -14,7 +14,11 @@ func (s *NoRebootNeeded) Success() {
 }
 
 func (s *NoRebootNeeded) Failure() {
-	s.stateful.SetState(&RebootNeeded{stateful: s.stateful})
+	if s.stateful.StreakUntilRebootState() > 1 {
+		s.stateful.SetState(NewUncertainState(s.stateful))
+	} else {
+		s.stateful.SetState(&RebootNeeded{stateful: s.stateful})
+	}
 }
 
 func (s *NoRebootNeeded) Error(err error) {