1 files changed, 15 insertions, 12 deletions
diff --git a/keyboards/westfoxtrot/aanzee/aanzee.c b/keyboards/westfoxtrot/aanzee/aanzee.c
index 0dc1199be9..17efdbe88d 100644
--- a/keyboards/westfoxtrot/aanzee/aanzee.c
+++ b/keyboards/westfoxtrot/aanzee/aanzee.c
@@ -30,23 +30,26 @@
 
 #include "aanzee.h"
 
+void keyboard_pre_init_kb(void) {
 
-void led_set_kb(uint8_t usb_led) {
-
-if (usb_led & (1<<USB_LED_CAPS_LOCK)) {
-
-// Turn capslock on
-
-  writePinLow(B2);
+    // Call the keyboard pre init code.
+    // Set our LED pins as output
+    setPinOutput(B2);
 
-} else {
+    keyboard_pre_init_user();
+}
 
-// Turn capslock off
+void led_set_kb(uint8_t usb_led) {
 
-  writePinHigh(B2);
+    if (IS_LED_ON(usb_led, USB_LED_CAPS_LOCK)) {
 
-}
+        // Turn capslock on
+        writePinLow(B2);
+    } else {
 
-led_set_user(usb_led);
+        // Turn capslock off
+        writePinHigh(B2);
+    }
 
+    led_set_user(usb_led);
 }