diff --git a/src/audio/stft_process/stft_process-generic.c b/src/audio/stft_process/stft_process-generic.c
index 5241c372261f..cbbe859b3e9a 100644
--- a/src/audio/stft_process/stft_process-generic.c
+++ b/src/audio/stft_process/stft_process-generic.c
@@ -4,7 +4,7 @@
 
 #include <sof/audio/format.h>
 #include <sof/common.h>
-#include <assert.h>
+#include <rtos/panic.h>
 #include <stdint.h>
 #include "stft_process.h"
 
diff --git a/src/audio/stft_process/stft_process-hifi3.c b/src/audio/stft_process/stft_process-hifi3.c
index 6cf7c3dc7e85..efcef00ca474 100644
--- a/src/audio/stft_process/stft_process-hifi3.c
+++ b/src/audio/stft_process/stft_process-hifi3.c
@@ -13,7 +13,7 @@
 
 #include <sof/audio/format.h>
 #include <sof/common.h>
-#include <assert.h>
+#include <rtos/panic.h>
 #include <stdint.h>
 #include "stft_process.h"