diff --git a/packages/applet/src/components/state/StateFieldEditor.vue b/packages/applet/src/components/state/StateFieldEditor.vue
index 93e7b91f..4010d138 100644
--- a/packages/applet/src/components/state/StateFieldEditor.vue
+++ b/packages/applet/src/components/state/StateFieldEditor.vue
@@ -3,9 +3,9 @@ import { computed, ref, toRaw } from 'vue'
 import { VueButton, VueDropdown, VueDropdownButton, VueIcon, vTooltip } from '@vue/devtools-ui'
 import { DevToolsV6PluginAPIHookKeys, getRaw } from '@vue/devtools-kit'
 import type { CustomInspectorState, DevToolsV6PluginAPIHookPayloads } from '@vue/devtools-kit'
-import type { ButtonProps } from '@vue/devtools-ui/dist/types/src/components/Button'
 import { rpc } from '@vue/devtools-core'
 import { useClipboard } from '@vueuse/core'
+import type { ButtonProps } from '@vue/devtools-ui'
 import { useStateEditorContext } from '~/composables/state-editor'
 import type { EditorAddNewPropType, EditorInputValidType } from '~/composables/state-editor'
 
diff --git a/packages/ui/package.json b/packages/ui/package.json
index ab808703..54184349 100644
--- a/packages/ui/package.json
+++ b/packages/ui/package.json
@@ -6,20 +6,20 @@
   "license": "MIT",
   "exports": {
     ".": {
-      "types": "./dist/types/src/index.d.ts",
+      "types": "./dist/types/index.d.ts",
       "import": "./dist/index.js",
       "require": "./dist/index.cjs"
     },
     "./style.css": "./dist/style.css",
     "./theme": {
-      "types": "./dist/types/theme/index.d.ts",
+      "types": "./dist/types/theme.d.ts",
       "import": "./dist/theme.js",
       "require": "./dist/theme.cjs"
     }
   },
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",
-  "types": "./dist/types/src/index.d.ts",
+  "types": "./dist/types/index.d.ts",
   "files": [
     "dist"
   ],
diff --git a/packages/ui/src/components/index.ts b/packages/ui/src/components/index.ts
index 6848e1b1..2dcaa1cc 100644
--- a/packages/ui/src/components/index.ts
+++ b/packages/ui/src/components/index.ts
@@ -39,3 +39,23 @@ export {
   Notification as VueNotification,
   Tooltip as VueTooltip,
 }
+
+export type * from './Button.vue'
+export type * from './Badge.vue'
+export type * from './LoadingIndicator.vue'
+export type * from './Icon.vue'
+export type * from './DarkToggle.vue'
+export type * from './Card.vue'
+export type * from './Dialog.vue'
+export type * from './Input.vue'
+export type * from './FormField.vue'
+export type * from './Dropdown.vue'
+export type * from './DropdownButton.vue'
+export type * from './Confirm.vue'
+export type * from './Select.vue'
+export type * from './Switch.vue'
+export type * from './Checkbox.vue'
+export type * from './Drawer.vue'
+export type * from './Overlay.vue'
+export type * from './Notification.vue'
+export type * from './Tooltip.vue'
diff --git a/packages/ui/tsconfig.json b/packages/ui/tsconfig.json
index 93cd89e3..d03814ee 100644
--- a/packages/ui/tsconfig.json
+++ b/packages/ui/tsconfig.json
@@ -4,7 +4,7 @@
     "jsx": "preserve",
     "lib": ["esnext", "DOM"],
     "module": "esnext",
-    "moduleResolution": "node",
+    "moduleResolution": "Bundler",
     "resolveJsonModule": true,
     "strict": true,
     "strictNullChecks": true,
diff --git a/packages/ui/vite.config.ts b/packages/ui/vite.config.ts
index 3e8989b2..bcbb680a 100644
--- a/packages/ui/vite.config.ts
+++ b/packages/ui/vite.config.ts
@@ -12,6 +12,7 @@ export default defineConfig({
     vue(),
     unocss(),
     dts({
+      rollupTypes: true,
       cleanVueFileName: true,
       include: ['src', 'theme'],
       outDir: 'dist/types',