diff --git a/js/display_atree.js b/js/display_atree.js
index 6e072cb..86d776b 100644
--- a/js/display_atree.js
+++ b/js/display_atree.js
@@ -87,7 +87,7 @@ function construct_AT(elem, tree) {
                 else {// if (parent_node.display.col < node.display.col && (parent_node.display.row != node.display.row)) {
                     connector.classList.add("rotate-270");
                 }
-                resolve_connector(document.getElementById("atree-row-" + parent_node.display.row).children[node.display.col]);
+                resolve_connector(document.getElementById("atree-row-" + parent_node.display.row).children[node.display.col], node);
             }
         }