diff --git a/packages/core/src/components/Edges/EdgeWrapper.ts b/packages/core/src/components/Edges/EdgeWrapper.ts
index 1a2f97b2..31220a14 100644
--- a/packages/core/src/components/Edges/EdgeWrapper.ts
+++ b/packages/core/src/components/Edges/EdgeWrapper.ts
@@ -189,6 +189,7 @@ const EdgeWrapper = defineComponent({
                       'centerX': sourceX,
                       'centerY': sourceY,
                       'radius': edgeUpdaterRadius.value,
+                      'type': 'source',
                       'data-type': 'source',
                     }),
                   ),
@@ -208,6 +209,7 @@ const EdgeWrapper = defineComponent({
                       'centerX': targetX,
                       'centerY': targetY,
                       'radius': edgeUpdaterRadius.value,
+                      'type': 'target',
                       'data-type': 'target',
                     }),
                   ),
diff --git a/packages/core/src/store/hooks.ts b/packages/core/src/store/hooks.ts
index e00693a7..e6b65a87 100644
--- a/packages/core/src/store/hooks.ts
+++ b/packages/core/src/store/hooks.ts
@@ -56,6 +56,7 @@ export function createHooks(): FlowHooks {
     edgeUpdateEnd: createExtendedEventHook(),
     updateNodeInternals: createExtendedEventHook(),
     error: createExtendedEventHook((err) => warn(err.message)),
+  }
 }
 
 export function useHooks(emit: (...args: any[]) => void, hooks: Ref<FlowHooks>) {