diff --git a/package/src/components/Edges/index.ts b/package/src/components/Edges/index.ts
index 4a004273..99e47b4e 100644
--- a/package/src/components/Edges/index.ts
+++ b/package/src/components/Edges/index.ts
@@ -6,3 +6,4 @@ export { default as SmoothStepEdge } from './SmoothStepEdge'
 export { default as StraightEdge } from './StraightEdge'
 export { default as EdgeAnchor } from './EdgeAnchor'
 export { default as EdgeText } from './EdgeText.vue'
+export { default as EdgeWrapper } from './EdgeWrapper.vue'
diff --git a/package/src/components/Nodes/index.ts b/package/src/components/Nodes/index.ts
index d638164e..f49bc08e 100644
--- a/package/src/components/Nodes/index.ts
+++ b/package/src/components/Nodes/index.ts
@@ -1,3 +1,4 @@
 export { default as DefaultNode } from './DefaultNode'
 export { default as InputNode } from './InputNode'
 export { default as OutputNode } from './OutputNode'
+export { default as NodeWrapper } from './NodeWrapper.vue'
diff --git a/package/src/components/index.ts b/package/src/components/index.ts
index d605cd6c..f491446e 100644
--- a/package/src/components/index.ts
+++ b/package/src/components/index.ts
@@ -1,6 +1,6 @@
 export * from './Nodes'
 export * from './Edges'
-export * from './ConnectionLine/ConnectionLine.vue'
-export * from './Handle/Handle.vue'
-export * from './NodesSelection/NodesSelection.vue'
-export * from './UserSelection/UserSelection.vue'
+export { default as ConnectionLine } from './ConnectionLine/ConnectionLine.vue'
+export { default as Handle } from './Handle/Handle.vue'
+export { default as NodesSelection } from './NodesSelection/NodesSelection.vue'
+export { default as UserSelection } from './UserSelection/UserSelection.vue'
diff --git a/package/src/container/EdgeRenderer/EdgeRenderer.vue b/package/src/container/EdgeRenderer/EdgeRenderer.vue
index f2dfdd47..919610ec 100644
--- a/package/src/container/EdgeRenderer/EdgeRenderer.vue
+++ b/package/src/container/EdgeRenderer/EdgeRenderer.vue
@@ -1,6 +1,5 @@
 <script lang="ts" setup>
-import EdgeWrapper from '../../components/Edges/EdgeWrapper.vue'
-import ConnectionLine from '../../components/ConnectionLine/ConnectionLine.vue'
+import { ConnectionLine, EdgeWrapper } from '../../components'
 import { useVueFlow } from '../../composables'
 import { groupEdgesByZLevel } from '../../utils'
 import type { EdgeComponent, GraphEdge } from '../../types'
diff --git a/package/src/container/NodeRenderer/NodeRenderer.vue b/package/src/container/NodeRenderer/NodeRenderer.vue
index 17db8706..7ab7adca 100644
--- a/package/src/container/NodeRenderer/NodeRenderer.vue
+++ b/package/src/container/NodeRenderer/NodeRenderer.vue
@@ -1,5 +1,5 @@
 <script lang="ts" setup>
-import NodeWrapper from '../../components/Nodes/NodeWrapper.vue'
+import { NodeWrapper } from '../../components'
 import type { GraphNode, NodeComponent, SnapGrid } from '../../types'
 import { useVueFlow } from '../../composables'
 import { Slots } from '../../context'
diff --git a/package/src/container/SelectionPane/SelectionPane.vue b/package/src/container/SelectionPane/SelectionPane.vue
index 4b2c625d..45d1bcf9 100644
--- a/package/src/container/SelectionPane/SelectionPane.vue
+++ b/package/src/container/SelectionPane/SelectionPane.vue
@@ -2,8 +2,7 @@
 import type { EdgeChange, NodeChange } from '../../types'
 import { useKeyPress, useVueFlow } from '../../composables'
 import { getConnectedEdges } from '../../utils'
-import NodesSelection from '../../components/NodesSelection/NodesSelection.vue'
-import UserSelection from '../../components/UserSelection/UserSelection.vue'
+import { NodesSelection, UserSelection } from '../../components'
 
 const {
   id,
diff --git a/package/src/container/index.ts b/package/src/container/index.ts
index a5913013..c000fc78 100644
--- a/package/src/container/index.ts
+++ b/package/src/container/index.ts
@@ -3,5 +3,5 @@ export { default as EdgeRenderer } from './EdgeRenderer/EdgeRenderer.vue'
 export { default as Marker } from './EdgeRenderer/Marker.vue'
 export { default as MarkerDefinitions } from './EdgeRenderer/MarkerDefinitions.vue'
 export { default as NodeRenderer } from './NodeRenderer/NodeRenderer.vue'
-export { default as ZoomPane } from './ZoomPane/ZoomPane.vue'
+export { default as Viewport } from './Viewport/Viewport.vue'
 export { default as SelectionPane } from './SelectionPane/SelectionPane.vue'
diff --git a/package/src/store/index.ts b/package/src/store/index.ts
index ef0a15d3..454d7f97 100644
--- a/package/src/store/index.ts
+++ b/package/src/store/index.ts
@@ -1,4 +1,4 @@
 export { default as useHooks, createHooks } from './hooks'
 export { default as useActions } from './actions'
-export { default as useState } from './state'
+export { default as useState, defaultEdgeTypes, defaultNodeTypes } from './state'
 export { default as useGetters } from './getters'