1 files changed, 11 insertions, 2 deletions
diff --git a/src/ast/nodes/ReturnStmtNode.java b/src/ast/nodes/ReturnStmtNode.java
index 15c8d69..1264fa9 100644
--- a/src/ast/nodes/ReturnStmtNode.java
+++ b/src/ast/nodes/ReturnStmtNode.java
@@ -63,15 +63,24 @@ public class ReturnStmtNode implements Node {
     }
 
     @Override
-    public String toPrint(String prefix) {
+    public String printAST(String prefix) {
         String str = prefix + "ReturnStmt\n";
 
         prefix += "  ";
         if (this.exprList != null) {
-            str += this.exprList.toPrint(prefix);
+            str += this.exprList.printAST(prefix);
         }
 
         return str;
     }
 
+    @Override
+    public String toPrint(String prefix) {
+        String str = prefix + "return ";
+        if (this.exprList != null) {
+            str += this.exprList.toPrint("");
+        }
+        return str + "\n";
+    }
+
 }