diff --git a/tests/runtime/custom_calyptia_input_test.c b/tests/runtime/custom_calyptia_input_test.c
index a09fe1c7c7d..25c05d25148 100644
--- a/tests/runtime/custom_calyptia_input_test.c
+++ b/tests/runtime/custom_calyptia_input_test.c
@@ -7,6 +7,7 @@
 #include <fluent-bit/flb_mem.h>
 #include <fluent-bit/flb_str.h>
 #include <fluent-bit/flb_utils.h>
+#include <fluent-bit/flb_input.h>
 #include "flb_tests_runtime.h"
 #include "../../plugins/custom_calyptia/calyptia.h"
 
diff --git a/tests/runtime/custom_calyptia_test.c b/tests/runtime/custom_calyptia_test.c
index 7a13fa48dd1..ca33b7efe06 100644
--- a/tests/runtime/custom_calyptia_test.c
+++ b/tests/runtime/custom_calyptia_test.c
@@ -2,6 +2,7 @@
 
 #include <fluent-bit.h>
 #include <fluent-bit/flb_scheduler.h>
+#include <fluent-bit/flb_input.h>
 #include <fluent-bit/flb_custom.h>
 #include "flb_tests_runtime.h"