Use statement writing for return statement in C# backend

1 files changed, 7 insertions, 7 deletions

diff --git a/src/main/java/org/zwobble/couscous/backends/csharp/CsharpSerializer.java b/src/main/java/org/zwobble/couscous/backends/csharp/CsharpSerializer.java
index c9df2ac..c7d6db1 100644
--- a/src/main/java/org/zwobble/couscous/backends/csharp/CsharpSerializer.java
+++ b/src/main/java/org/zwobble/couscous/backends/csharp/CsharpSerializer.java
@@ -160,10 +160,12 @@ public class CsharpSerializer implements NodeVisitor {
 
     @Override
     public void visit(ReturnNode returnNode) {
-        writer.writeKeyword("return");
-        writer.writeSpace();
-        write(returnNode.getValue());
-        writer.writeSymbol(";");
+        writer.writeStatement(() -> {
+            writer.writeKeyword("return");
+            writer.writeSpace();
+            write(returnNode.getValue());
+            writer.writeSymbol(";");
+        });
     }
 
     @Override
@@ -220,9 +222,7 @@ public class CsharpSerializer implements NodeVisitor {
                 () -> writer.writeSymbol(", "));
             writer.writeSymbol(")");
             writer.startBlock();
-            for (StatementNode statement : method.getBody()) {
-                writer.writeStatement(() -> write(statement));
-            }
+            writeAll(method.getBody());
             writer.endBlock();
         });
     }