Have Patio.return_() accept an operand instead of a string

2 files changed, 3 insertions, 3 deletions

diff --git a/src/test/java/org/shedlang/compiler/patio/PatioSerializerTests.java b/src/test/java/org/shedlang/compiler/patio/PatioSerializerTests.java
index b84b794b..7034196e 100644
--- a/src/test/java/org/shedlang/compiler/patio/PatioSerializerTests.java
+++ b/src/test/java/org/shedlang/compiler/patio/PatioSerializerTests.java
@@ -411,7 +411,7 @@ public class PatioSerializerTests {
 
     @Test
     public void canSerializeReturn() {
-        var instruction = Patio.return_("value");
+        var instruction = Patio.return_(Patio.localRef("value"));
 
         var string = PatioTesting.serializeInstruction(instruction);
 
diff --git a/src/test/java/org/shedlang/compiler/patio/instructions/Patio.java b/src/test/java/org/shedlang/compiler/patio/instructions/Patio.java
index 01a82ce2..9de86434 100644
--- a/src/test/java/org/shedlang/compiler/patio/instructions/Patio.java
+++ b/src/test/java/org/shedlang/compiler/patio/instructions/Patio.java
@@ -213,7 +213,7 @@ public class Patio {
         );
     }
 
-    public static PatioInstruction return_(String value) {
-        return new PatioReturn(new PatioLocalRef(value), Sources.NULL);
+    public static PatioInstruction return_(PatioOperand value) {
+        return new PatioReturn(value, Sources.NULL);
     }
 }