diff --git a/test/libei-unit-test.c b/test/libei-unit-test.c
index a2db95d..8f15afb 100644
--- a/test/libei-unit-test.c
+++ b/test/libei-unit-test.c
@@ -5,6 +5,8 @@
 #include <sys/socket.h>
 #include <munit.h>
 
+#include "util-mem.h"
+
 extern const MunitSuite __start_test_section, __stop_test_section;
 
 int
@@ -12,7 +14,7 @@ main(int argc, char* argv[MUNIT_ARRAY_PARAM(argc + 1)])
 {
 	size_t sz = 10;
 	size_t count = 0;
-	MunitSuite *suites = calloc(sz, sizeof(*suites));
+	_cleanup_free_ MunitSuite *suites = calloc(sz, sizeof(*suites));
 
 	for (const MunitSuite *s = &__start_test_section;
 	     s < &__stop_test_section;