2 files changed, 7 insertions, 5 deletions
diff --git a/chapter10a/src/main/java/com/seaofnodes/simple/node/ReturnNode.java b/chapter10a/src/main/java/com/seaofnodes/simple/node/ReturnNode.java
index f13f3708..96d58599 100644
--- a/chapter10a/src/main/java/com/seaofnodes/simple/node/ReturnNode.java
+++ b/chapter10a/src/main/java/com/seaofnodes/simple/node/ReturnNode.java
@@ -4,15 +4,16 @@ import com.seaofnodes.simple.type.*;
 
 import java.util.BitSet;
 
-/** Keeps the return value and all preceding memory effects alive. */
+/** Keeps all preceding memory effects and the return value alive: {ctrl, $mem, value}. */
 public class ReturnNode extends Node {
 
     public ReturnNode(Node ctrl, Node data, ScopeNode scope) {
-        super(ctrl, data, scope.lookup("$mem"));
+        super(ctrl, scope.lookup("$mem"), data);
     }
 
     public Node ctrl() { return in(0); }
-    public Node expr() { return in(1); }
+    public Node mem () { return in(1); }
+    public Node expr() { return in(2); }
 
     @Override
     public String label() { return "Return"; }
@@ -28,7 +29,8 @@ public class ReturnNode extends Node {
 
     @Override
     public Type compute() {
-        return TypeTuple.make(ctrl()._type,expr()._type);
+        // Return exposes the complete memory state.
+        return TypeTuple.make(ctrl()._type,TypeMem.BOT,expr()._type);
     }
 
     @Override
diff --git a/chapter10a/src/main/java/com/seaofnodes/simple/print/SimpleGraphAdapter.java b/chapter10a/src/main/java/com/seaofnodes/simple/print/SimpleGraphAdapter.java
index 5e21c852..600e9c73 100644
--- a/chapter10a/src/main/java/com/seaofnodes/simple/print/SimpleGraphAdapter.java
+++ b/chapter10a/src/main/java/com/seaofnodes/simple/print/SimpleGraphAdapter.java
@@ -52,7 +52,7 @@ public class SimpleGraphAdapter extends GraphAdapter<Node> {
         if( n instanceof ProjNode ) return n.isCFG() ? Role.CTRL : isMem(n) ? Role.MEM : Role.DATA;
         if( n instanceof RegionNode && i == 0 ) return Role.ASSOC;
         if( i == 0 || n instanceof RegionNode || n instanceof StopNode ) return Role.CTRL;
-        if( i == 1 && n instanceof MemOpNode || i == 2 && n instanceof ReturnNode ) return Role.MEM;
+        if( i == 1 && (n instanceof MemOpNode || n instanceof ReturnNode) ) return Role.MEM;
         Node def = n.in(i);
         if( def != null && isMem(def) ) return Role.MEM;
         return Role.DATA;