diff --git a/firmware/boards/f0_module/bootloader/Makefile b/firmware/boards/f0_module/bootloader/Makefile
index ca6f8e7..577dc94 100644
--- a/firmware/boards/f0_module/bootloader/Makefile
+++ b/firmware/boards/f0_module/bootloader/Makefile
@@ -8,6 +8,8 @@ ifeq ($(USE_OPT),)
   USE_OPT = -Os -ggdb -fomit-frame-pointer -falign-functions=16 -fsingle-precision-constant
 endif
 
+USE_OPT += -DWB_PROD=1
+
 # C specific options here (added to USE_OPT).
 ifeq ($(USE_COPT),)
   USE_COPT = 
diff --git a/firmware/boards/f0_module/bootloader/bootloader.cpp b/firmware/boards/f0_module/bootloader/bootloader.cpp
index 24ad38f..0dc630a 100644
--- a/firmware/boards/f0_module/bootloader/bootloader.cpp
+++ b/firmware/boards/f0_module/bootloader/bootloader.cpp
@@ -1,3 +1,5 @@
+#include "ch.h"
+#include "hal.h"
 
 #include "port_shared.h"
 #include "flash.h"