diff --git a/docs/components/examples/stress/App.vue b/docs/components/examples/stress/App.vue
index 6be2fb8d..931fea0d 100644
--- a/docs/components/examples/stress/App.vue
+++ b/docs/components/examples/stress/App.vue
@@ -30,12 +30,6 @@ const updatePos = () => {
     fitView({ duration: 1000, padding: 0.5 })
   })
 }
-
-onNodeClick(({ connectedEdges }) => {
-  connectedEdges.forEach((edge) => {
-    edge.style = { stroke: '#10b981', strokeWidth: 3 }
-  })
-})
 </script>
 
 <template>
diff --git a/docs/components/examples/stress/utils.js b/docs/components/examples/stress/utils.js
index 4eccda32..60a7869b 100644
--- a/docs/components/examples/stress/utils.js
+++ b/docs/components/examples/stress/utils.js
@@ -17,7 +17,15 @@ export function getElements(xElements = 10, yElements = 10) {
       initialNodes.push(node)
 
       if (recentNodeId && nodeId <= xElements * yElements) {
-        initialEdges.push({ id: `${x}-${y}`, source: recentNodeId.toString(), target: nodeId.toString() })
+        initialEdges.push({
+          id: `${x}-${y}`,
+          source: recentNodeId.toString(),
+          target: nodeId.toString(),
+          style: (edge) => {
+            if (!edge.sourceNode.selected && !edge.targetNode.selected) return
+            return { stroke: '#10b981', strokeWidth: 3 }
+          },
+        })
       }
 
       recentNodeId = nodeId