CFG fix create edges only if nodes != (found by graphviz)

1 files changed, 6 insertions, 3 deletions

diff --git a/vaporize/library/CFGNode.java b/vaporize/library/CFGNode.java
index 0afce60..b79ee6b 100644
--- a/vaporize/library/CFGNode.java
+++ b/vaporize/library/CFGNode.java
@@ -33,9 +33,12 @@ class CFGNode {
      */
     // FIXME
     public boolean equals(Object other) {
-        return (other instanceof Node &&
-                (((Node) other).sourcePos ==
-                 this.instruction.sourcePos)) ||
+        return (other instanceof CFGNode &&
+                (((CFGNode) other).instruction ==
+                 this.instruction)) ||
+            (other instanceof Node &&
+             (((Node) other).sourcePos ==
+              this.instruction.sourcePos)) ||
             (other instanceof Integer &&
              (((Integer) other)
               .equals(this.instruction.sourcePos.line)));