diff --git a/app/boards/shields/sofle_rgb_keyhive/boards/nice_nano_v2.overlay b/app/boards/shields/sofle_rgb_keyhive/boards/nice_nano_v2.overlay
index fe6f5670..765b5552 100644
--- a/app/boards/shields/sofle_rgb_keyhive/boards/nice_nano_v2.overlay
+++ b/app/boards/shields/sofle_rgb_keyhive/boards/nice_nano_v2.overlay
@@ -1,3 +1,5 @@
+#include <dt-bindings/led/led.h>
+
 /*
  * Copyright (c) 2022 The ZMK Contributors
  *
@@ -24,6 +26,8 @@
     chain-length = <16>; /* number of LEDs */
     spi-one-frame = <0x70>;
     spi-zero-frame = <0x40>;
+
+    color-mapping = <LED_COLOR_ID_GREEN LED_COLOR_ID_RED LED_COLOR_ID_BLUE>;
   };
 };
 
diff --git a/app/boards/shields/sofle_rgb_keyhive/sofle_rgb_keyhive.dtsi b/app/boards/shields/sofle_rgb_keyhive/sofle_rgb_keyhive.dtsi
index 0bce38a9..7b1e35e9 100644
--- a/app/boards/shields/sofle_rgb_keyhive/sofle_rgb_keyhive.dtsi
+++ b/app/boards/shields/sofle_rgb_keyhive/sofle_rgb_keyhive.dtsi
@@ -8,6 +8,7 @@
 
 / {
     chosen {
+        zephyr,display = &oled;
         zmk,kscan = &kscan0;
         zmk,matrix_transform = &default_transform;
     };
@@ -71,7 +72,7 @@ RC(3,5) RC(3,4) RC(3,3) RC(3,2) RC(3,1) RC(3,0) RC(4,5)  RC(4,6) RC(3,6) RC(3,7)
 &pro_micro_i2c {
     status = "okay";
 
-    ssd1306@3c {
+    oled: ssd1306@3c {
         compatible = "solomon,ssd1306fb";
         reg = <0x3c>;
         label = "DISPLAY";