From 076cd1d6c40fece8702ef00396c7008176377f4c Mon Sep 17 00:00:00 2001
From: Daniele Lacamera <dan@danielinux.net>
Date: Tue, 4 Mar 2025 18:20:46 +0100
Subject: [PATCH] Fix for unit test failing

---
 include/image.h                  | 2 ++
 tools/unit-tests/unit-extflash.c | 1 +
 2 files changed, 3 insertions(+)

diff --git a/include/image.h b/include/image.h
index 02d7ae27..c2f44839 100644
--- a/include/image.h
+++ b/include/image.h
@@ -89,7 +89,9 @@ int wolfBot_get_dts_size(void *dts_addr);
     #else
         #define XALIGNED(x) /* null expansion */
     #endif
+#endif
 
+#ifndef XALIGNED_STACK
     /* Don't enforce stack alignment on IAR */
     #if defined (__IAR_SYSTEMS_ICC__)
         #define XALIGNED_STACK(x)
diff --git a/tools/unit-tests/unit-extflash.c b/tools/unit-tests/unit-extflash.c
index d698b344..1dfbd51a 100644
--- a/tools/unit-tests/unit-extflash.c
+++ b/tools/unit-tests/unit-extflash.c
@@ -47,6 +47,7 @@
 #include <stdlib.h>
 #include <string.h>
 #include "user_settings.h"
+#include "image.h"
 
 #include "libwolfboot.c"