diff --git a/docs/components/examples/edges/CustomEdge.vue b/docs/components/examples/edges/CustomEdge.vue
index 52fe1385..65934731 100644
--- a/docs/components/examples/edges/CustomEdge.vue
+++ b/docs/components/examples/edges/CustomEdge.vue
@@ -45,12 +45,12 @@ const props = defineProps({
   },
 })
 
-const { applyEdgeChanges } = useVueFlow()
+const { removeEdges } = useVueFlow()
 
 const foreignObjectSize = 40
 
 const onClick = (evt, id) => {
-  applyEdgeChanges([{ type: 'remove', id }])
+  removeEdges([id])
   evt.stopPropagation()
 }