1 files changed, 9 insertions, 6 deletions
diff --git a/keyboards/4pplet/waffling60/rev_c/rev_c.c b/keyboards/4pplet/waffling60/rev_c/rev_c.c
index f00cce06dc..fc7e99b601 100644
--- a/keyboards/4pplet/waffling60/rev_c/rev_c.c
+++ b/keyboards/4pplet/waffling60/rev_c/rev_c.c
@@ -19,12 +19,15 @@ 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 && res) {
-        if (led_state.caps_lock){
-            rgblight_enable();
-            rgblight_mode(1);
-            rgblight_sethsv_noeeprom(CAPS_LOCK_COLOR);
-        } else {
-            rgblight_disable_noeeprom();
+        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;