diff --git a/tests/scripts/generate_code.pl b/tests/scripts/generate_code.pl
index 75d741d..4565c61 100755
--- a/tests/scripts/generate_code.pl
+++ b/tests/scripts/generate_code.pl
@@ -59,6 +59,12 @@
 {
 $suite_pre_code
 
+#if defined(POLARSSL_MEMORY_BUFFER_ALLOC_C)
+    unsigned char buf[1000000];
+    memory_buffer_alloc_init( buf, sizeof(buf) );
+#endif
+
+
     FCT_SUITE_BGN($suite_name)
     {
 END
@@ -141,6 +147,10 @@
     }
     FCT_SUITE_END();
 
+#if defined(POLARSSL_MEMORY_BUFFER_ALLOC_C) && defined(POLARSSL_MEMORY_DEBUG)
+    memory_buffer_alloc_status();
+#endif
+
 $suite_post_code
 }
 FCT_END();