From 2e7d1bb669e3ee26efe060ca5ea92ef73cebff83 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sat, 6 Feb 2016 17:38:26 +0000
Subject: Add return type to MethodSignature

---
 src/main/java/org/zwobble/couscous/ast/MethodNode.java | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'src/main/java/org/zwobble/couscous/ast/MethodNode.java')

diff --git a/src/main/java/org/zwobble/couscous/ast/MethodNode.java b/src/main/java/org/zwobble/couscous/ast/MethodNode.java
index e0b2432..dd970c5 100644
--- a/src/main/java/org/zwobble/couscous/ast/MethodNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/MethodNode.java
@@ -140,7 +140,8 @@ public class MethodNode implements CallableNode {
     public MethodSignature signature() {
         return new MethodSignature(
             name,
-            eagerMap(arguments, argument -> argument.getType()));
+            eagerMap(arguments, argument -> argument.getType()),
+            returnType);
     }
 
     public MethodNode mapBody(Function<List<StatementNode>, List<StatementNode>> function) {
-- 
cgit v1.2.3