diff --git a/ports/stm32/mboot/main.c b/ports/stm32/mboot/main.c
index caaf394687..da4463e1cc 100644
--- a/ports/stm32/mboot/main.c
+++ b/ports/stm32/mboot/main.c
@@ -441,10 +441,7 @@ MP_WEAK void led_init(void) {
     led0_cur_state = LED0_STATE_OFF;
 }
 
-MP_WEAK void led_state(int led, int val) {
-    if (led == 1) {
-        led = LED0;
-    }
+MP_WEAK void led_state(uint32_t led, int val) {
     if (val) {
         MICROPY_HW_LED_ON(led);
     } else {