1 files changed, 2 insertions, 2 deletions
diff --git a/script/size.py b/script/size.py
index e1f6605..c11e4cd 100755
--- a/script/size.py
+++ b/script/size.py
@@ -31,8 +31,8 @@ def main(size_executable, rom_sections, ram_sections):
             section_size[section] = size
 
     total = {
-        "ROM": sum(map(lambda section: section_size[section], rom_sections)),
-        "RAM": sum(map(lambda section: section_size[section], ram_sections)),
+        "ROM": sum(map(lambda section: section_size.get(section, 0), rom_sections)),
+        "RAM": sum(map(lambda section: section_size.get(section, 0), ram_sections)),
     }
 
     output = {"section": section_size, "total": total}