diff --git a/stm-firmware/main.c b/stm-firmware/main.c
index 63a73f8..b60a97e 100644
--- a/stm-firmware/main.c
+++ b/stm-firmware/main.c
@@ -164,6 +164,11 @@ static inline void handle_boot_status(void)
 
 		start_updater_ram_code();
 	}
+
+	if (status.code_updated) {
+		status.code_updated = 0x0UL;
+		safety_memory_set_boot_status(&status);
+	}
 }
 
 static inline void setup_system(void)