diff --git a/python/cuxfilter/charts/core/non_aggregate/core_graph.py b/python/cuxfilter/charts/core/non_aggregate/core_graph.py
index af07e87b..ab9b4101 100644
--- a/python/cuxfilter/charts/core/non_aggregate/core_graph.py
+++ b/python/cuxfilter/charts/core/non_aggregate/core_graph.py
@@ -50,14 +50,14 @@ def edge_columns(self):
 
     @property
     def node_columns(self):
+        cols = [
+            self.node_id,
+            self.node_x,
+            self.node_y,
+        ]
         if self.node_aggregate_col != self.node_id:
-            return [
-                self.node_id,
-                self.node_x,
-                self.node_y,
-                self.node_aggregate_col,
-            ]
-        return [self.node_id, self.node_x, self.node_y]
+            cols.append(self.node_aggregate_col)
+        return cols
 
     def __init__(
         self,