diff --git a/test/simple-memory-corruption/invalid_malloc_object_size_small.c b/test/simple-memory-corruption/invalid_malloc_object_size_small.c
index f1796a3..126ff2b 100644
--- a/test/simple-memory-corruption/invalid_malloc_object_size_small.c
+++ b/test/simple-memory-corruption/invalid_malloc_object_size_small.c
@@ -1,9 +1,10 @@
 #include <malloc.h>
 
+#include "../test_util.h"
+
 size_t malloc_object_size(void *ptr);
 
-__attribute__((optimize(0)))
-int main() {
+OPTNONE int main(void) {
     char *p = malloc(16);
     if (!p) {
         return 1;
diff --git a/test/simple-memory-corruption/invalid_malloc_object_size_small_quarantine.c b/test/simple-memory-corruption/invalid_malloc_object_size_small_quarantine.c
index ac8a7e8..168dc23 100644
--- a/test/simple-memory-corruption/invalid_malloc_object_size_small_quarantine.c
+++ b/test/simple-memory-corruption/invalid_malloc_object_size_small_quarantine.c
@@ -1,9 +1,10 @@
 #include <malloc.h>
 
+#include "../test_util.h"
+
 size_t malloc_object_size(void *ptr);
 
-__attribute__((optimize(0)))
-int main() {
+OPTNONE int main(void) {
     void *p = malloc(16);
     if (!p) {
         return 1;