From db85ba86e4dd7c10abd7918313ea114285284084 Mon Sep 17 00:00:00 2001
From: Brian Fjeldstad <bfjelds@microsoft.com>
Date: Wed, 12 Nov 2025 19:49:33 +0000
Subject: [PATCH] allow rerun of update that has failed health check

---
 crates/trident/src/lib.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/crates/trident/src/lib.rs b/crates/trident/src/lib.rs
index fb5f6da64..5a9bc4613 100644
--- a/crates/trident/src/lib.rs
+++ b/crates/trident/src/lib.rs
@@ -604,7 +604,7 @@ impl Trident {
                             Ok(ExitKind::Done)
                         }
                     }
-                    ServicingState::AbUpdateFinalized | ServicingState::Provisioned => {
+                    ServicingState::AbUpdateHealthCheckFailed | ServicingState::AbUpdateFinalized | ServicingState::Provisioned => {
                         // Need to either re-execute the failed update OR inform the user that no update
                         // is needed.
                         engine::update(&host_config, datastore, &allowed_operations, image, #[cfg(feature = "grpc-dangerous")] sender).message("Failed to update host")