Add return type to MethodNode

1 files changed, 2 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/frontends/java/JavaReader.java b/src/main/java/org/zwobble/couscous/frontends/java/JavaReader.java
index a436bd7..f9cfd85 100644
--- a/src/main/java/org/zwobble/couscous/frontends/java/JavaReader.java
+++ b/src/main/java/org/zwobble/couscous/frontends/java/JavaReader.java
@@ -100,6 +100,7 @@ public class JavaReader {
             false,
             functionalInterfaceMethod.getName(),
             lambda.getFormalArguments(),
+            typeOf(functionalInterfaceMethod.getReturnType()),
             lambda.getBody());
 
         return new AnonymousClass(
@@ -313,6 +314,7 @@ public class JavaReader {
                 Modifier.isStatic(method.getModifiers()),
                 method.getName().getIdentifier(),
                 formalArguments,
+                typeOf(method.getReturnType2()),
                 body);
         }
     }