diff --git a/packages/core/src/components/Edges/StepEdge.ts b/packages/core/src/components/Edges/StepEdge.ts
index 8f85e33a..3d99c499 100644
--- a/packages/core/src/components/Edges/StepEdge.ts
+++ b/packages/core/src/components/Edges/StepEdge.ts
@@ -1,7 +1,8 @@
 import { defineComponent, h } from 'vue'
 import SmoothStepEdge from './SmoothStepEdge'
+import type { StepEdgeProps } from '~/types'
 
-const StepEdge = defineComponent({
+const StepEdge = defineComponent<StepEdgeProps>({
   name: 'StepEdge',
   props: [
     'sourcePosition',