diff --git a/app/src/behaviors/behavior_tri_state.c b/app/src/behaviors/behavior_tri_state.c
index 1338e058..463694e4 100644
--- a/app/src/behaviors/behavior_tri_state.c
+++ b/app/src/behaviors/behavior_tri_state.c
@@ -6,9 +6,9 @@
 
 #define DT_DRV_COMPAT zmk_behavior_tri_state
 
-#include <device.h>
+#include <zephyr/device.h>
 #include <drivers/behavior.h>
-#include <logging/log.h>
+#include <zephyr/logging/log.h>
 #include <zmk/behavior.h>
 #include <zmk/behavior_queue.h>
 #include <zmk/keymap.h>