diff --git a/tests/static_addr_taken/lib.c b/tests/static_addr_taken/lib.c
index 8d532fb18..389f0e130 100644
--- a/tests/static_addr_taken/lib.c
+++ b/tests/static_addr_taken/lib.c
@@ -1,5 +1,4 @@
-#include <criterion/criterion.h>
-#include <criterion/logging.h>
+#include <ia2_test_runner.h>
 #include <ia2.h>
 
 #define IA2_COMPARTMENT 2
@@ -22,4 +21,4 @@ static fn_ptr_ty ptrs[3] IA2_SHARED_DATA = {
 
 fn_ptr_ty *get_ptrs_in_lib(void) {
     return ptrs;
-}
\ No newline at end of file
+}
diff --git a/tests/static_addr_taken/main.c b/tests/static_addr_taken/main.c
index 4db4c951e..57b6ba744 100644
--- a/tests/static_addr_taken/main.c
+++ b/tests/static_addr_taken/main.c
@@ -1,5 +1,4 @@
-#include <criterion/criterion.h>
-#include <criterion/logging.h>
+#include <ia2_test_runner.h>
 #include <ia2.h>
 
 INIT_RUNTIME(2);
@@ -40,4 +39,4 @@ Test(static_addr_taken, call_ptr_from_lib) {
     for (int i = 0; i < 3; i++) {
         lib_ptrs[i]();
     }
-}
\ No newline at end of file
+}