diff --git a/keyboards/annepro2/annepro2.c b/keyboards/annepro2/annepro2.c
index 42523acae0d..87d78ff317b 100644
--- a/keyboards/annepro2/annepro2.c
+++ b/keyboards/annepro2/annepro2.c
@@ -2,6 +2,7 @@
 #include "hal.h"
 #include "annepro2.h"
 #include "annepro2_ble.h"
+#include "qmk_ap2_led.h"
 
 static const SerialConfig ledUartConfig = {
   .speed = 115200,
@@ -52,21 +53,45 @@ bool OVERRIDE process_record_kb(uint16_t keycode, keyrecord_t *record) {
             case KC_AP2_BT1:
                 annepro2_ble_broadcast(0);
                 return false;
+
             case KC_AP2_BT2:
                 annepro2_ble_broadcast(1);
                 return false;
+
             case KC_AP2_BT3:
                 annepro2_ble_broadcast(2);
                 return false;
+
             case KC_AP2_BT4:
                 annepro2_ble_broadcast(3);
                 return false;
+
             case KC_AP2_USB:
                 annepro2_ble_disconnect();
                 return false;
+
             case KC_AP2_BT_UNPAIR:
                 annepro2_ble_unpair();
                 return false;
+
+            case KC_AP_LED_OFF:
+                annepro2LedDisable();
+                annepro2LedPrevProfile();
+                return false;
+
+            case KC_AP_LED_ON:
+                annepro2LedNextProfile();
+                annepro2LedEnable();
+                return false;
+
+            case KC_AP_LED_NEXT_PROFILE:
+                annepro2LedNextProfile();
+                return false;
+
+            case KC_AP_LED_PREV_PROFILE:
+               annepro2LedPrevProfile();
+               return false;
+
             default:
                 break;
         }
diff --git a/keyboards/annepro2/annepro2.h b/keyboards/annepro2/annepro2.h
index 8c6b58fcf40..1a81861147c 100644
--- a/keyboards/annepro2/annepro2.h
+++ b/keyboards/annepro2/annepro2.h
@@ -16,7 +16,7 @@
  */
 
 #pragma once
-
+#include "annepro2.h"
 #include "quantum.h"
 
 #define OVERRIDE __attribute__ ((noinline))
@@ -32,3 +32,10 @@ enum AP2KeyCodes {
     KC_AP2_USB,
     AP2_SAFE_RANGE,
 };
+
+enum custom_keys {
+    KC_AP_LED_ON = AP2_SAFE_RANGE,
+    KC_AP_LED_OFF,
+    KC_AP_LED_NEXT_PROFILE,
+    KC_AP_LED_PREV_PROFILE
+};
diff --git a/keyboards/annepro2/keymaps/default/keymap.c b/keyboards/annepro2/keymaps/default/keymap.c
index e4818e5d256..a5053d6563e 100644
--- a/keyboards/annepro2/keymaps/default/keymap.c
+++ b/keyboards/annepro2/keymaps/default/keymap.c
@@ -7,12 +7,7 @@ enum anne_pro_layers {
   _FN1_LAYER,
   _FN2_LAYER,
 };
-enum custom_keys {
-    KC_AP_LED_ON = AP2_SAFE_RANGE,
-    KC_AP_LED_OFF,
-    KC_AP_LED_NEXT_PROFILE,
-    KC_AP_LED_PREV_PROFILE 
-};
+
 /*
 * Layer _BASE_LAYER
 * ,-----------------------------------------------------------------------------------------.
@@ -105,34 +100,3 @@ void matrix_scan_user(void) {
 layer_state_t layer_state_set_user(layer_state_t layer) {
     return layer;
 }
-
-/*!
- * @returns false   processing for this keycode has been completed.
- */
-bool process_record_user(uint16_t keycode, keyrecord_t *record) {
-    switch (keycode) {
-        case KC_AP_LED_OFF:
-            if (record->event.pressed) {
-                annepro2LedDisable();
-                annepro2LedPrevProfile();
-            }
-            return false;
-        case KC_AP_LED_ON:
-            if (record->event.pressed) {
-                annepro2LedEnable();
-                annepro2LedNextProfile();
-            }
-            return false;
-	case KC_AP_LED_NEXT_PROFILE:
-	    if (record->event.pressed) 
-	        annepro2LedNextProfile();
-	    return false;
-	case KC_AP_LED_PREV_PROFILE:
-	    if (record->event.pressed)
-	        annepro2LedPrevProfile();
-	    return false;
-        default:
-            break;
-    }
-    return true;
-}