diff --git a/src/lib-zephyr/cpu.c b/src/lib-zephyr/cpu.c
index c9b307907..4c29f7389 100644
--- a/src/lib-zephyr/cpu.c
+++ b/src/lib-zephyr/cpu.c
@@ -123,6 +123,7 @@ int cpu_enabled_cores(void)
 
 #else /* CONFIG_ACE */
 #include <sof/trace/trace.h>
+#include <rtos/wait.h>
 #include <zephyr/pm/pm.h>
 
 LOG_MODULE_DECLARE(zephyr, CONFIG_SOF_LOG_LEVEL);