From cfbb047e9778f42ea70ec6abfcb268d9377c3e85 Mon Sep 17 00:00:00 2001
From: Aetias <aetias@outlook.com>
Date: Sun, 15 Oct 2023 18:02:34 +0200
Subject: [PATCH] Write ROM size

---
 tools/rom/build.c | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tools/rom/build.c b/tools/rom/build.c
index d1ea3ca3..bd5bf893 100644
--- a/tools/rom/build.c
+++ b/tools/rom/build.c
@@ -640,7 +640,8 @@ int main(int argc, const char **argv) {
 		return 1;
 	}
 	
-    header.capacity = 15 - __builtin_clz(address);
+    header.romSize = address;
+    header.capacity = 15 - __builtin_clz(header.romSize);
 	size_t romEnd = 1 << (17 + header.capacity);
 	if (!Align(romEnd, fpRom, &address)) return 1;