diff --git a/chip/lm4/pwm_kblight.c b/chip/lm4/pwm_kblight.c
index 9382a8e35f..8710896928 100644
--- a/chip/lm4/pwm_kblight.c
+++ b/chip/lm4/pwm_kblight.c
@@ -11,6 +11,7 @@
 #include "gpio.h"
 #include "hooks.h"
 #include "host_command.h"
+#include "lid_switch.h"
 #include "pwm.h"
 #include "registers.h"
 #include "system.h"
@@ -188,3 +189,9 @@ static void pwm_kblight_shutdown(void)
 	pwm_set_keyboard_backlight(0);
 }
 DECLARE_HOOK(HOOK_CHIPSET_SHUTDOWN, pwm_kblight_shutdown, HOOK_PRIO_DEFAULT);
+
+static void pwm_kblight_lid_change(void)
+{
+	pwm_enable_keyboard_backlight(lid_is_open());
+}
+DECLARE_HOOK(HOOK_LID_CHANGE, pwm_kblight_lid_change, HOOK_PRIO_DEFAULT);
diff --git a/chip/lm4/switch.c b/chip/lm4/switch.c
index c4f1526275..168b57eb7c 100644
--- a/chip/lm4/switch.c
+++ b/chip/lm4/switch.c
@@ -153,11 +153,6 @@ static void update_backlight(void)
 		gpio_set_level(GPIO_ENABLE_BACKLIGHT, 1);
 	else
 		gpio_set_level(GPIO_ENABLE_BACKLIGHT, 0);
-
-#ifdef CONFIG_PWM_KBLIGHT
-	/* Same with keyboard backlight */
-	pwm_enable_keyboard_backlight(lid_is_open());
-#endif
 }
 
 /**