Parse try expression as branching expression

1 files changed, 6 insertions, 1 deletions

diff --git a/src/main/java/org/shedlang/compiler/ast/untyped/UntypedTryNode.java b/src/main/java/org/shedlang/compiler/ast/untyped/UntypedTryNode.java
index 95e85a61..70a7e2bb 100644
--- a/src/main/java/org/shedlang/compiler/ast/untyped/UntypedTryNode.java
+++ b/src/main/java/org/shedlang/compiler/ast/untyped/UntypedTryNode.java
@@ -12,7 +12,12 @@ public record UntypedTryNode(
     UntypedTypeLevelExpressionNode handledEffect,
     List<UntypedTryHandlerMethodNode> handlerMethods,
     Source source
-) implements UntypedExpressionNode {
+) implements UntypedBranchingExpressionNode, UntypedExpressionNode {
+    @Override
+    public Iterable<UntypedBlockNode> exhaustiveBranchBodies() {
+        return List.of(body);
+    }
+
     public static Builder builder() {
         return new Builder(
             P.vector(),