From 5a7665a9e3462ecd81c606e8e5cfdea84a281d21 Mon Sep 17 00:00:00 2001
From: John Bland <john.bland@wolfssl.com>
Date: Tue, 22 Aug 2023 22:40:41 -0400
Subject: [PATCH] add missing FINAL_SWAP check

---
 src/update_flash.c | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/update_flash.c b/src/update_flash.c
index dc0e4600..bd2c1a19 100644
--- a/src/update_flash.c
+++ b/src/update_flash.c
@@ -853,7 +853,7 @@ void RAMFUNCTION wolfBoot_start(void)
          * wolfBoot_update_trigger is called and thus it shouldn't be called
          * if an update is already in progress */
 #if defined(NVM_FLASH_WRITEONCE) || defined(WOLFBOOT_FLAGS_INVERT)
-        if ((wolfBoot_get_partition_state(PART_UPDATE, &st) != 0) || (st != IMG_STATE_UPDATING))
+        if ((wolfBoot_get_partition_state(PART_UPDATE, &st) != 0) || (st != IMG_STATE_UPDATING && st != IMG_STATE_FINAL_SWAP))
 #endif
         {
             wolfBoot_update_trigger();