1 files changed, 6 insertions, 4 deletions
diff --git a/src/main/java/org/shedlang/compiler/backends/wasm/PatioToWasm.java b/src/main/java/org/shedlang/compiler/backends/wasm/PatioToWasm.java
index c4e73446..707d9a87 100644
--- a/src/main/java/org/shedlang/compiler/backends/wasm/PatioToWasm.java
+++ b/src/main/java/org/shedlang/compiler/backends/wasm/PatioToWasm.java
@@ -361,6 +361,7 @@ public class PatioToWasm {
 
             case PatioSwitch patioSwitch -> {
                 // TODO: use br_table?
+
                 var wasmBlockTypeResult = generateBlockType(patioSwitch.destinations(), context);
                 var wasmBlockType = wasmBlockTypeResult.value();
                 context = wasmBlockTypeResult.context();
@@ -399,10 +400,11 @@ public class PatioToWasm {
             }
 
             case PatioTry try_ -> {
-                // TODO: don't assume we always have a Patio int
-                context = context.enterTryInstruction(
-                    WasmBlockType.of(target.patioIntType())
-                );
+                var wasmBlockTypeResult = generateBlockType(try_.destinations(), context);
+                var wasmBlockType = wasmBlockTypeResult.value();
+                context = wasmBlockTypeResult.context();
+
+                context = context.enterTryInstruction(wasmBlockType);
 
                 context = generateBlockBody(try_.body(), context);