1 files changed, 17 insertions, 13 deletions

diff --git a/keyboards/xbows/knight/knight.c b/keyboards/xbows/knight/knight.c
index 1b8adc8c2..f697f23f9 100644
--- a/keyboards/xbows/knight/knight.c
+++ b/keyboards/xbows/knight/knight.c
@@ -1,19 +1,23 @@
 #include "knight.h"
+
 void matrix_init_kb(void) {
+    // put your keyboard start-up code here
+    // runs once when the firmware starts up
+
+    matrix_init_user();
+    led_init_ports();
+}
+
+void led_init_ports(void) {
     setPinOutput(D1);
+    writePinHigh(D1);
     setPinOutput(E2);
-    matrix_init_user();
-};
-void led_set_kb(uint8_t usb_led) {
-    if (IS_LED_ON(usb_led, USB_LED_NUM_LOCK)) {
-        writePinHigh(D1);
-    } else {
-        writePinLow(D1);
-    }
-    if (IS_LED_ON(usb_led, USB_LED_CAPS_LOCK)) {
-        writePinHigh(E2);
-    } else {
-        writePinLow(E2);
+    writePinHigh(E2);
+}
+bool led_update_kb(led_t led_state) {
+    if(led_update_user(led_state)) {
+        writePin(E2, !led_state.caps_lock);
+        writePin(D1, !led_state.num_lock);
     }
-    led_set_user(usb_led);
+    return true;
 }