From 26a68e83ed8af30616b682ab3c7fb826742c381b Mon Sep 17 00:00:00 2001
From: braks <78412429+bcakmakoglu@users.noreply.github.com>
Date: Fri, 24 Mar 2023 22:40:26 +0100
Subject: [PATCH] chore(core): cleanup

Signed-off-by: braks <78412429+bcakmakoglu@users.noreply.github.com>
---
 packages/core/src/components/Edges/EdgeWrapper.ts | 2 ++
 packages/core/src/store/hooks.ts                  | 1 +
 2 files changed, 3 insertions(+)

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>) {