diff --git a/examples/graph_visual_with_neo4j.py b/examples/graph_visual_with_neo4j.py
index 7377f21c..35ef85a8 100644
--- a/examples/graph_visual_with_neo4j.py
+++ b/examples/graph_visual_with_neo4j.py
@@ -62,7 +62,7 @@ def main():
         e.displayName = node.id
     REMOVE e:Entity
     WITH e, node
-    CALL apoc.create.addLabels(e, [node.entity_type]) YIELD node AS labeledNode
+    CALL apoc.create.addLabels(e, [node.id]) YIELD node AS labeledNode
     RETURN count(*)
     """