Change method bodies to be optional

2 files changed, 2 insertions, 2 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 37147ed..3cfcd33 100644
--- a/src/main/java/org/zwobble/couscous/backends/csharp/CsharpSerializer.java
+++ b/src/main/java/org/zwobble/couscous/backends/csharp/CsharpSerializer.java
@@ -265,7 +265,7 @@ public class CsharpSerializer implements NodeVisitor {
             writer.writeSpace();
             writer.writeIdentifier(method.getName());
             writeFormalArguments(method.getArguments());
-            writeBlock(method.getBody());
+            writeBlock(method.getBody().get());
         });
     }
 
diff --git a/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java b/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java
index 2aafcca..08ecba4 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java
@@ -166,7 +166,7 @@ public class PythonCodeGenerator {
         Iterable<String> argumentNames = Iterables.concat(
             method.isStatic() ? Collections.<String>emptyList() : list("self"),
             explicitArgumentNames);
-        List<PythonStatementNode> pythonBody = generateStatements(method.getBody());
+        List<PythonStatementNode> pythonBody = generateStatements(method.getBody().get());
         return pythonFunctionDefinition(Names.toUniqueName(method.signature()), ImmutableList.copyOf(argumentNames), new PythonBlock(pythonBody));
     }