1 files changed, 4 insertions, 2 deletions

diff --git a/keyboards/rubi/lib/oled.c b/keyboards/rubi/lib/oled.c
index d48d27323..9534cdd9b 100644
--- a/keyboards/rubi/lib/oled.c
+++ b/keyboards/rubi/lib/oled.c
@@ -239,7 +239,8 @@ void render_frame(void) {
     }
 }
 
-__attribute__((weak)) void oled_task_user(void) {
+bool oled_task_kb(void) {
+    if (!oled_task_user()) { return false; }
     if (timer_elapsed(oled_frame_timer) > OLED_FRAME_TIMEOUT) {
         oled_clear();
         oled_frame_timer = timer_read();
@@ -259,9 +260,10 @@ __attribute__((weak)) void oled_task_user(void) {
     } else {
         oled_mode = OLED_MODE_DEFAULT;
     }
+    return false;
 }
 
-oled_rotation_t oled_init_user(oled_rotation_t rotation) {
+oled_rotation_t oled_init_kb(oled_rotation_t rotation) {
     oled_logo_timer = timer_read();
     oled_frame_timer = timer_read();
     return rotation;