1 files changed, 4 insertions, 0 deletions
diff --git a/src/test/java/org/shedlang/compiler/patio/PatioSerializerTests.java b/src/test/java/org/shedlang/compiler/patio/PatioSerializerTests.java
index f9b2116e..59bbdaaf 100644
--- a/src/test/java/org/shedlang/compiler/patio/PatioSerializerTests.java
+++ b/src/test/java/org/shedlang/compiler/patio/PatioSerializerTests.java
@@ -17,6 +17,7 @@ public class PatioSerializerTests {
     public void canSerializeCompilationUnit() {
         var element = PatioFunctionDefinition.builder()
             .withName("f")
+            .withReturnType(PatioTypes.VOID)
             .build();
         var compilationUnit = Patio.compilationUnit(List.of(element));
 
@@ -30,6 +31,7 @@ public class PatioSerializerTests {
                     is-private=true
                     params=()
                     locals=()
+                    return-type=void
                     body=(
                     )
                 )
@@ -50,6 +52,7 @@ public class PatioSerializerTests {
             .addParam(Patio.local("param2", PatioTypes.DATA_POINTER))
             .addLocal(Patio.local("local1", PatioTypes.FUNCTION_POINTER))
             .addLocal(Patio.local("local2", PatioTypes.INT))
+            .withReturnType(PatioTypes.VOID)
             .addBodyInstruction(Patio.intGet("local2", 2))
             .build();
 
@@ -62,6 +65,7 @@ public class PatioSerializerTests {
                 is-private=true
                 params=(("param1" int) ("param2" data-ptr))
                 locals=(("local1" function-ptr) ("local2" int))
+                return-type=void
                 body=(
                     (int-get "local2" 2)
                 )