diff --git a/keyboards/ergodox_ez/keymaps/taudris/keymap.c b/keyboards/ergodox_ez/keymaps/taudris/keymap.c
index 22060bc1ef..8cc980798f 100644
--- a/keyboards/ergodox_ez/keymaps/taudris/keymap.c
+++ b/keyboards/ergodox_ez/keymaps/taudris/keymap.c
@@ -279,7 +279,7 @@ bool process_record_user(uint16_t keycode, keyrecord_t *record) {
 #endif
       //turn off capslock when pressing shift to type a shifted key
       if (
-        host_keyboard_leds() & (1<<USB_LED_CAPS_LOCK) && //capslock on
+        host_keyboard_led_state().caps_lock &&
         (
 			temp_mod & MODS_SHIFT_MASK
 #ifndef NO_ACTION_ONESHOT
@@ -382,10 +382,11 @@ layer_state_t layer_state_set_user(layer_state_t state) {
   return state;
 };
 
-void led_set_kb(uint8_t usb_led) {
-    if (usb_led & (1<<USB_LED_CAPS_LOCK)) {
+bool led_update_user(led_t led_state) {
+    if (led_state.caps_lock) {
         ergodox_right_led_1_on();
     } else {
         ergodox_right_led_1_off();
     }
+    return true;
 }