diff --git a/tests/lib/custom_lib/src/main.c b/tests/lib/custom_lib/src/main.c
index 16c1681..25477e2 100644
--- a/tests/lib/custom_lib/src/main.c
+++ b/tests/lib/custom_lib/src/main.c
@@ -11,8 +11,8 @@
  * library works correctly.
  */
 
-#include <zephyr.h>
-#include <ztest.h>
+#include <zephyr/kernel.h>
+#include <zephyr/ztest.h>
 #include <custom_lib/custom_lib.h>
 
 static void test_get_value(void)