2 files changed, 7 insertions, 2 deletions
diff --git a/keyboards/marksard/treadstone48/common/oled_helper.c b/keyboards/marksard/treadstone48/common/oled_helper.c
index 68adbe83a8..c3c8f23cbc 100644
--- a/keyboards/marksard/treadstone48/common/oled_helper.c
+++ b/keyboards/marksard/treadstone48/common/oled_helper.c
@@ -1,8 +1,10 @@
-#ifdef OLED_ENABLE
-#include QMK_KEYBOARD_H
+#include "oled_helper.h"
+#include "quantum.h"
 #include <stdio.h>
 #include <string.h>
 
+#ifdef OLED_ENABLE
+
 void render_logo(void) {
 
   const char logo_buf[]={
diff --git a/keyboards/marksard/treadstone48/common/oled_helper.h b/keyboards/marksard/treadstone48/common/oled_helper.h
index 56c2a5b236..82c1feba4a 100644
--- a/keyboards/marksard/treadstone48/common/oled_helper.h
+++ b/keyboards/marksard/treadstone48/common/oled_helper.h
@@ -1,5 +1,8 @@
 #ifdef OLED_ENABLE
 
+#include <stdint.h>
+#include "action.h"
+
 void render_logo(void);
 void update_key_status(uint16_t keycode, keyrecord_t *record);
 void render_key_status(void);