Further fix for #121

1 files changed, 7 insertions, 0 deletions

diff --git a/src/codegen/codegen_decl.c b/src/codegen/codegen_decl.c
index b82e1af..d525963 100644
--- a/src/codegen/codegen_decl.c
+++ b/src/codegen/codegen_decl.c
@@ -991,7 +991,14 @@ int emit_tests_and_runner(ParserContext *ctx, ASTNode *node, FILE *out)
         if (cur->type == NODE_TEST)
         {
             fprintf(out, "static void _z_test_%d() {\n", test_count);
+            int saved = defer_count;
             codegen_walker(ctx, cur->test_stmt.body, out);
+            // Run defers
+            for (int i = defer_count - 1; i >= saved; i--)
+            {
+                codegen_node_single(ctx, defer_stack[i], out);
+            }
+            defer_count = saved;
             fprintf(out, "}\n");
             test_count++;
         }