2 files changed, 8 insertions, 1 deletions
diff --git a/src/test/java/org/shedlang/compiler/stackir/StackCompilerFunctionDefinitionTests.java b/src/test/java/org/shedlang/compiler/stackir/StackCompilerFunctionDefinitionTests.java
index 1fa6c41a..1c90a0a7 100644
--- a/src/test/java/org/shedlang/compiler/stackir/StackCompilerFunctionDefinitionTests.java
+++ b/src/test/java/org/shedlang/compiler/stackir/StackCompilerFunctionDefinitionTests.java
@@ -36,7 +36,8 @@ public class StackCompilerFunctionDefinitionTests {
                 has("freeVariables", x -> x.freeVariables(), containsExactly(equalTo("z"))),
                 has("body", x -> x.body(), isSequence(
                     isPushStringInstruction("hello"),
-                    isDiscardInstruction()
+                    isDiscardInstruction(),
+                    isPushUnitInstruction()
                 ))
             ),
             isLocalStoreInstruction("f")
diff --git a/src/test/java/org/shedlang/compiler/stackir/instructions/InstructionMatchers.java b/src/test/java/org/shedlang/compiler/stackir/instructions/InstructionMatchers.java
index bfac50ce..0cbad43d 100644
--- a/src/test/java/org/shedlang/compiler/stackir/instructions/InstructionMatchers.java
+++ b/src/test/java/org/shedlang/compiler/stackir/instructions/InstructionMatchers.java
@@ -73,4 +73,10 @@ public class InstructionMatchers {
             has("value", x -> x.value(), equalTo(value))
         );
     }
+
+    public static Matcher<Instruction> isPushUnitInstruction() {
+        return instanceOf(
+            PushUnitInstruction.class
+        );
+    }
 }