diff --git a/examples/src/Basic/Basic.vue b/examples/src/Basic/Basic.vue
index 529ed115..7ba916aa 100644
--- a/examples/src/Basic/Basic.vue
+++ b/examples/src/Basic/Basic.vue
@@ -1,5 +1,5 @@
 <script lang="ts" setup>
-import { VueFlow, MiniMap, Controls, Background, isNode, useVueFlow, Elements } from '@braks/vue-flow'
+import { VueFlow, MiniMap, Controls, Background, isNode, useVueFlow, Elements } from '@braks/vue-flow/src/index'
 
 const elements = ref<Elements>([
   { id: '1', type: 'input', label: 'Node 1', position: { x: 250, y: 5 } },
diff --git a/package/src/utils/graph.ts b/package/src/utils/graph.ts
index e1661ed7..dff6ecc2 100644
--- a/package/src/utils/graph.ts
+++ b/package/src/utils/graph.ts
@@ -7,6 +7,7 @@ import {
   Edge,
   EdgeMarkerType,
   Elements,
+  FlowElement,
   FlowElements,
   FlowExportObject,
   Getters,
@@ -52,14 +53,13 @@ export const getHostForElement = (element: HTMLElement): Document => {
   else return window.document
 }
 
-export const isEdge = (element: Node | Edge | Connection): element is Edge =>
-  'id' in element && 'source' in element && 'target' in element
-export const isGraphEdge = (element: any): element is GraphEdge =>
+type MaybeElement = Node | Edge | Connection | FlowElement
+export const isEdge = (element: MaybeElement): element is Edge => 'id' in element && 'source' in element && 'target' in element
+export const isGraphEdge = (element: MaybeElement): element is GraphEdge =>
   isEdge(element) && 'sourceNode' in element && 'targetNode' in element
 
-export const isNode = (element: Node | Edge | Connection): element is Node => 'id' in element && !isEdge(element)
-export const isGraphNode = (element: Node | Edge | Connection): element is GraphNode =>
-  isNode(element) && 'computedPosition' in element
+export const isNode = (element: MaybeElement): element is Node => 'id' in element && !isEdge(element)
+export const isGraphNode = (element: MaybeElement): element is GraphNode => isNode(element) && 'computedPosition' in element
 
 export const parseNode = (node: Node, nodeExtent: CoordinateExtent, defaults?: Partial<GraphNode>): GraphNode => {
   let defaultValues = defaults