diff --git a/red-teaming/markOwnedNodesInNeo4j.py b/red-teaming/markOwnedNodesInNeo4j.py
index 4c8eb3f..b3ea186 100644
--- a/red-teaming/markOwnedNodesInNeo4j.py
+++ b/red-teaming/markOwnedNodesInNeo4j.py
@@ -47,8 +47,9 @@ def markNodes(tx, nodes):
 
     for i in range(len(nodes)):
         n = nodes[i]
-        query += 'MATCH (n {name: "' + n + '"}) SET n.owned = TRUE '
-        if i < len(nodes) - 1: query += 'UNION '
+        query += 'MATCH (n {name: "' + n + '"}) SET n.owned=TRUE RETURN 1'
+        if i < len(nodes) - 1: query += ' UNION'
+        query += '\n'
 
     tx.run(query)