diff --git a/packages/svelte/src/lib/components/EdgeWrapper/EdgeWrapper.svelte b/packages/svelte/src/lib/components/EdgeWrapper/EdgeWrapper.svelte
index e22f5c89..0b99d744 100644
--- a/packages/svelte/src/lib/components/EdgeWrapper/EdgeWrapper.svelte
+++ b/packages/svelte/src/lib/components/EdgeWrapper/EdgeWrapper.svelte
@@ -57,7 +57,7 @@
 <!-- svelte-ignore a11y-no-noninteractive-element-interactions -->
 {#if !hidden}
   <g
-    class="svelte-flow__edge"
+    class="svelte-flow__edge {className}"
     class:animated
     class:selected
     data-id={id}
@@ -71,7 +71,6 @@
   >
     <svelte:component
       this={edgeComponent}
-      class={className}
       {id}
       {source}
       {target}
diff --git a/packages/svelte/src/lib/types/edges.ts b/packages/svelte/src/lib/types/edges.ts
index 19e7e47f..c4426a89 100644
--- a/packages/svelte/src/lib/types/edges.ts
+++ b/packages/svelte/src/lib/types/edges.ts
@@ -69,6 +69,7 @@ export type EdgeLayouted = Pick<
   | 'targetHandle'
   | 'ariaLabel'
   | 'hidden'
+  | 'class'
 > &
   EdgePosition & {
     sourceHandleId?: string | null;