2 files changed, 6 insertions, 2 deletions
diff --git a/keyboards/marksard/rhymestone/common/oled_helper.c b/keyboards/marksard/rhymestone/common/oled_helper.c
index 613798c941..4a0ca72e89 100644
--- a/keyboards/marksard/rhymestone/common/oled_helper.c
+++ b/keyboards/marksard/rhymestone/common/oled_helper.c
@@ -1,9 +1,11 @@
 #include "oled_helper.h"
-#include "quantum.h"
+#include "host.h"
+#include "progmem.h"
 #include <stdio.h>
 #include <string.h>
 
 #ifdef OLED_ENABLE
+#include "oled_driver.h"
 
 void render_logo(void) {
 
diff --git a/keyboards/marksard/treadstone48/common/oled_helper.c b/keyboards/marksard/treadstone48/common/oled_helper.c
index e9a8cf564a..66124abe12 100644
--- a/keyboards/marksard/treadstone48/common/oled_helper.c
+++ b/keyboards/marksard/treadstone48/common/oled_helper.c
@@ -1,5 +1,7 @@
 #include "oled_helper.h"
-#include "quantum.h"
+#include "oled_driver.h"
+#include "host.h"
+#include "rgblight.h"
 #include <stdio.h>
 #include <string.h>