Introduce TypedFunctionBodyNode

1 files changed, 2 insertions, 2 deletions

diff --git a/src/main/java/org/shedlang/compiler/stackir/StackCompiler.java b/src/main/java/org/shedlang/compiler/stackir/StackCompiler.java
index 33d84149..2bd1229a 100644
--- a/src/main/java/org/shedlang/compiler/stackir/StackCompiler.java
+++ b/src/main/java/org/shedlang/compiler/stackir/StackCompiler.java
@@ -203,7 +203,7 @@ public class StackCompiler {
             ))
             .toList();
 
-        context = compileBlockWithYield(node.body(), context);
+        context = compileBlockWithYield(node.body().block(), context);
 
         switch (node.body().behaviour()) {
             case TypedFunctionStatementBehaviour.BlockReturn blockReturn -> {
@@ -221,7 +221,7 @@ public class StackCompiler {
         context = context.exitBlock();
 
         var freeVariables = FreeVariables.builder();
-        for (var shedFreeVariable : node.freeVariables()) {
+        for (var shedFreeVariable : node.body().freeVariables()) {
             var freeVariableLocalName = context.shedVariableToLocal(shedFreeVariable.name());
             for (var freeVariableSource : shedFreeVariable.referenceSources()) {
                 freeVariables = freeVariables.plus(