diff --git a/packages/core/src/context/index.ts b/packages/core/src/context/index.ts
index 3c5e4d56..df9c4e1e 100644
--- a/packages/core/src/context/index.ts
+++ b/packages/core/src/context/index.ts
@@ -5,5 +5,5 @@ export const VueFlow: InjectionKey<VueFlowStore> = Symbol('vueFlow')
 export const NodeId: InjectionKey<string> = Symbol('nodeId')
 export const NodeRef: InjectionKey<Ref<HTMLDivElement>> = Symbol('nodeRef')
 export const EdgeId: InjectionKey<string> = Symbol('edgeId')
-export const EdgeRef: InjectionKey<Ref<HTMLDivElement>> = Symbol('edgeRef')
+export const EdgeRef: InjectionKey<Ref<SVGElement>> = Symbol('edgeRef')
 export const Slots: InjectionKey<TSlots> = Symbol('slots')