CFG captures both VBranch paths

1 files changed, 6 insertions, 3 deletions

diff --git a/vaporize/library/ControlFlowGraph.java b/vaporize/library/ControlFlowGraph.java
index e7057ac..c0eff10 100644
--- a/vaporize/library/ControlFlowGraph.java
+++ b/vaporize/library/ControlFlowGraph.java
@@ -15,7 +15,7 @@ public class ControlFlowGraph {
         this.start = null;
     }
 
-    protected CFGNode getNode(Node a) {
+    protected CFGNode getNode(Object a) {
         CFGNode ret = null;
         for (CFGNode n : this.nodes) {
             if (n.equals(a)) {
@@ -24,9 +24,12 @@ public class ControlFlowGraph {
             }
         }
 
-        if (ret == null)
+        if (ret == null) {
+            String str = (a instanceof Node) ? ((Node) a).sourcePos.toString() :
+                a.toString();
             MinimalLogger.severe(String.format("Could not find a node matching %s",
-                                               a.sourcePos.toString()));
+                                               str));
+        }
         return ret;
     }