diff --git a/docs/components/DocsRepl.vue b/docs/components/DocsRepl.vue
index 58e0e207..65051d64 100644
--- a/docs/components/DocsRepl.vue
+++ b/docs/components/DocsRepl.vue
@@ -74,6 +74,7 @@ onMounted(async () => {
     imports: {
       '@vue-flow/additional-components': `${location.origin}/vue-flow-additional-components.mjs`,
       '@vue-flow/core': `${location.origin}/vue-flow-core.mjs`,
+      '@vue-flow/node-toolbar': `${location.origin}/vue-flow-node-toolbar.mjs`,
       ...additionalImports,
     },
   })
diff --git a/docs/components/examples/node-toolbar/App.vue b/docs/components/examples/node-toolbar/App.vue
index cf2637a1..962e4f6d 100644
--- a/docs/components/examples/node-toolbar/App.vue
+++ b/docs/components/examples/node-toolbar/App.vue
@@ -1,5 +1,6 @@
 <script setup>
 import { Position, VueFlow } from '@vue-flow/core'
+import { ref } from 'vue'
 import ToolbarNode from './ToolbarNode.vue'
 
 const defaultNodeStyle = {
diff --git a/docs/src/components.d.ts b/docs/src/components.d.ts
index 48884bc7..d1a7b145 100644
--- a/docs/src/components.d.ts
+++ b/docs/src/components.d.ts
@@ -38,6 +38,7 @@ declare module '@vue/runtime-core' {
     SnappableConnectionLine: typeof import('./../components/examples/snap-to-handle/SnappableConnectionLine.vue')['default']
     Team: typeof import('./../components/home/Team.vue')['default']
     TeleportableNode: typeof import('./../components/examples/teleport/TeleportableNode.vue')['default']
+    ToolbarNode: typeof import('./../components/examples/node-toolbar/ToolbarNode.vue')['default']
     TransitionEdge: typeof import('./../components/examples/transition/TransitionEdge.vue')['default']
   }
 }