diff --git a/src/main.c b/src/main.c
index 04b21e3..3c8b6d6 100644
--- a/src/main.c
+++ b/src/main.c
@@ -195,7 +195,7 @@ static void arena_test_allocator_interface(void)
     MemoryZero(ptr, page_size);
     printf("test_4: allocated %u bytes page-aligned at %p\n", page_size, ptr);
 
-    arena_free(&arena);
+    allocator_free_all(&allocator);
 }
 
 static void arena_test_decommit(void)
@@ -306,7 +306,7 @@ void apool_test(void)
     for (PoolTest *p = list; p; p = p->next){
         printf("Pooltest %u\n",p->n);
     }
-    arena_free(&arena);
+    allocator_free_all(&allocator);
 }
 
 void apool_test_autogrow(void)
@@ -354,6 +354,7 @@ void apool_test_autogrow(void)
         printf("Pooltest %u\n",p->n);
     }
 
+    allocator_free_all(&allocator);
 }
 
 int main(int argc, char *argv[])