diff --git a/examples/EdgeWithButton/EdgeWithButton.vue b/examples/EdgeWithButton/EdgeWithButton.vue
index 56a6b833..bc5d804c 100644
--- a/examples/EdgeWithButton/EdgeWithButton.vue
+++ b/examples/EdgeWithButton/EdgeWithButton.vue
@@ -31,8 +31,7 @@ const elements = ref<Elements>([
 
 const onLoad = (flowInstance: FlowInstance) => flowInstance.fitView()
 const onElementsRemove = (elementsToRemove: Elements) => (elements.value = removeElements(elementsToRemove, elements.value))
-const onConnect = (params: Connection | Edge) =>
-  (elements.value = addEdge({ ...params, type: 'buttonedge' } as Edge, elements.value))
+const onConnect = (params: Connection | Edge) => (elements.value = addEdge({ ...params, type: 'button' } as Edge, elements.value))
 </script>
 <template>
   <VueFlow