diff --git a/rewriter/tests/rewrite_macros/CMakeLists.txt b/rewriter/tests/rewrite_macros/CMakeLists.txt
index 3cf926439d..d995917889 100644
--- a/rewriter/tests/rewrite_macros/CMakeLists.txt
+++ b/rewriter/tests/rewrite_macros/CMakeLists.txt
@@ -5,6 +5,7 @@ define_shared_lib(
 define_test(
     SRCS main.c
     NEEDS_LD_WRAP
+    CRITERION_TEST
 )
 
 define_ia2_wrapper()
diff --git a/rewriter/tests/rewrite_macros/main.c b/rewriter/tests/rewrite_macros/main.c
index b31333b52d..75ca0df857 100644
--- a/rewriter/tests/rewrite_macros/main.c
+++ b/rewriter/tests/rewrite_macros/main.c
@@ -4,12 +4,13 @@ RUN: cat main.c | FileCheck --match-full-lines --check-prefix=REWRITER %s
 #include "lib.h"
 #include <stddef.h>
 #include <ia2.h>
+#include <criterion/criterion.h>
 
 INIT_RUNTIME(1);
 #define IA2_COMPARTMENT 1
 #include <ia2_compartment_init.inc>
 
-int main() {
+Test(rewrite_macros, main) {
     init_actions();
 
     // The next two lines use macros that cannot be easily rewritten so the