diff --git a/apps/client/.storybook/preview.tsx b/apps/client/.storybook/preview.tsx
index 3377efcb..38e1b811 100644
--- a/apps/client/.storybook/preview.tsx
+++ b/apps/client/.storybook/preview.tsx
@@ -1,6 +1,6 @@
 import { Global, ThemeProvider } from '@emotion/react';
 import type { Preview } from '@storybook/react';
-import { theme } from '@tiki/ui/dist/index.mjs';
+import { theme } from '@tiki/ui';
 
 import React from 'react';
 import { MemoryRouter } from 'react-router-dom';
diff --git a/apps/client/vite.config.ts b/apps/client/vite.config.ts
index f207d9ff..841a878a 100644
--- a/apps/client/vite.config.ts
+++ b/apps/client/vite.config.ts
@@ -38,8 +38,8 @@ export default defineConfig(({ mode }) => {
 
     resolve: {
       alias: {
-        '@tiki/ui': path.resolve(__dirname, '../../packages/ui'),
-        '@tiki/icon': path.resolve(__dirname, '../../packages/icon'),
+        '@tiki/ui': path.resolve(__dirname, '../../packages/ui/dist'),
+        '@tiki/icon': path.resolve(__dirname, '../../packages/icon/dist'),
       },
     },