Change method bodies to be optional

1 files changed, 3 insertions, 3 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/MethodNode.java b/src/main/java/org/zwobble/couscous/ast/MethodNode.java
index 967b2b1..c201047 100644
--- a/src/main/java/org/zwobble/couscous/ast/MethodNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/MethodNode.java
@@ -11,7 +11,7 @@ import java.util.function.Function;
 
 import static org.zwobble.couscous.util.ExtraLists.eagerMap;
 
-public class MethodNode implements CallableNode {
+public class MethodNode implements Node {
     public static Builder staticMethod(String name) {
         return builder(name).isStatic(true);
     }
@@ -140,8 +140,8 @@ public class MethodNode implements CallableNode {
         return returnType;
     }
     
-    public List<StatementNode> getBody() {
-        return body.get();
+    public Optional<List<StatementNode>> getBody() {
+        return body;
     }
 
     public MethodSignature signature() {