diff --git a/src/include/sof/audio/kpb.h b/src/include/sof/audio/kpb.h
index b9575022a..45a9d692f 100644
--- a/src/include/sof/audio/kpb.h
+++ b/src/include/sof/audio/kpb.h
@@ -131,4 +131,8 @@ struct comp_data {
 	struct comp_buffer *rt_sink; /**< real time sink (channel selector ) */
 };
 
+#ifdef UNIT_TEST
+void sys_comp_kpb_init(void);
+#endif
+
 #endif
diff --git a/test/cmocka/src/audio/kpb/kpb_mock.c b/test/cmocka/src/audio/kpb/kpb_mock.c
index 5cda63f04..fcc768a20 100644
--- a/test/cmocka/src/audio/kpb/kpb_mock.c
+++ b/test/cmocka/src/audio/kpb/kpb_mock.c
@@ -5,6 +5,7 @@
 #include <cmocka.h>
 
 #include <sof/alloc.h>
+#include <sof/notifier.h>
 #include <sof/audio/component.h>
 #include <mock_trace.h>