diff --git a/tools/unit-tests/unit-update-disk.c b/tools/unit-tests/unit-update-disk.c
index a80acafb80..b694cdbc03 100644
--- a/tools/unit-tests/unit-update-disk.c
+++ b/tools/unit-tests/unit-update-disk.c
@@ -1,6 +1,7 @@
 #define WOLFBOOT_UPDATE_DISK
 #define WOLFBOOT_SKIP_BOOT_VERIFY
 #define WOLFBOOT_SELF_UPDATE_MONOLITHIC
+#define RAM_CODE
 #define WOLFBOOT_SELF_HEADER
 #define EXT_ENCRYPTED
 #define ENCRYPT_WITH_CHACHA