1 files changed, 10 insertions, 9 deletions
diff --git a/keyboards/4pplet/waffling60/rev_b/rev_b.c b/keyboards/4pplet/waffling60/rev_b/rev_b.c
index 5f701b57f4..c03f3630e6 100644
--- a/keyboards/4pplet/waffling60/rev_b/rev_b.c
+++ b/keyboards/4pplet/waffling60/rev_b/rev_b.c
@@ -18,15 +18,16 @@ along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
 bool led_update_kb(led_t led_state) {
     bool res = led_update_user(led_state);
-    if(CAPS_LOCK_ENABLE){
-        if(res) {
-            if(led_state.caps_lock){
-                rgblight_enable();
-                rgblight_mode(1);
-                rgblight_sethsv_noeeprom(CAPS_LOCK_COLOR);
-            }
-            else
-                rgblight_disable_noeeprom();
+    if (CAPS_LOCK_ENABLE && res) {
+        if(led_state.caps_lock) {
+            #ifdef CAPS_LOCK_COLOR
+                rgblight_sethsv_at(CAPS_LOCK_COLOR, 0);
+            #else
+                rgblight_sethsv_at(rgblight_get_hue(),rgblight_get_sat(),rgblight_get_val(), 0);
+            #endif
+        }
+        else{
+            rgblight_sethsv_at(HSV_OFF, 0);
         }
     }
     return res;