diff --git a/packages/docs/.storybook/main.ts b/packages/docs/.storybook/main.ts
index f3d201d..cb52b6f 100644
--- a/packages/docs/.storybook/main.ts
+++ b/packages/docs/.storybook/main.ts
@@ -28,5 +28,11 @@ const config: StorybookConfig = {
   docs: {
     autodocs: true,
   },
+  viteFinal: (config, { configType }) => {
+    if (configType === 'PRODUCTION') {
+      config.base = "/design-system/";
+    }
+    return config;
+  }
 };
 export default config;