diff --git a/src/components/Edges/EdgeWrapper.vue b/src/components/Edges/EdgeWrapper.vue
index 693921d4..9875b2be 100644
--- a/src/components/Edges/EdgeWrapper.vue
+++ b/src/components/Edges/EdgeWrapper.vue
@@ -55,7 +55,7 @@ const handleEdgeUpdater = (event: MouseEvent, isSourceHandle: boolean) => {
     isValidConnection,
     isSourceHandle ? 'target' : 'source',
     (connection) => store.hooks.edgeUpdate.trigger({ edge: props.edge, connection }),
-    () => store.hooks.edgeUpdateEnd.trigger(event),
+    () => store.hooks.edgeUpdateEnd.trigger({ event, edge: props.edge }),
   )
 }
 
diff --git a/src/composables/useHandle.ts b/src/composables/useHandle.ts
index 2294a3de..d71a29d9 100644
--- a/src/composables/useHandle.ts
+++ b/src/composables/useHandle.ts
@@ -153,10 +153,7 @@ export default (store = useStore()) =>
 
       store.hooks.connectEnd.trigger(event)
 
-      if (elementEdgeUpdaterType) {
-        store.hooks.edgeUpdateEnd.trigger(event)
-        onEdgeUpdateEnd?.()
-      }
+      if (elementEdgeUpdaterType) onEdgeUpdateEnd?.()
 
       resetRecentHandle(recentHoveredHandle)
       store.setConnectionNodeId({ connectionNodeId: undefined, connectionHandleId: undefined, connectionHandleType: undefined })
diff --git a/src/types/hooks.ts b/src/types/hooks.ts
index 44aa06f8..7e40a3cc 100644
--- a/src/types/hooks.ts
+++ b/src/types/hooks.ts
@@ -45,7 +45,7 @@ export interface FlowEvents {
   edgeDoubleClick: { event: MouseEvent; edge: GraphEdge }
   edgeClick: { event: MouseEvent; edge: GraphEdge }
   edgeUpdateStart: { event: MouseEvent; edge: GraphEdge }
-  edgeUpdateEnd: MouseEvent
+  edgeUpdateEnd: { event: MouseEvent; edge: GraphEdge }
 }
 
 export type FlowHooks = { [key in keyof FlowEvents]: FlowHook<FlowEvents[key]> }