diff --git a/packages/components/.storybook/main.ts b/packages/components/.storybook/main.ts
index 00bb790a..155d61dc 100644
--- a/packages/components/.storybook/main.ts
+++ b/packages/components/.storybook/main.ts
@@ -1,4 +1,5 @@
 import type { StorybookConfig } from "@storybook/react-vite";
+import * as path from "path";
 
 const config: StorybookConfig = {
     stories: [
@@ -11,10 +12,15 @@ const config: StorybookConfig = {
         "@storybook/addon-onboarding",
         "@storybook/addon-interactions",
     ],
-    framework: {
-        name: "@storybook/react-vite",
-        options: {},
-    },
+    framework: path.resolve(
+        require.resolve("@storybook/react-vite"),
+        "..",
+    ) as any,
+    // This doesn't work with Yarn workspaces, so we use the above workaround instead.
+    // framework: {
+    //     name: "@storybook/react-vite",
+    //     options: {},
+    // },
     docs: {
         autodocs: "tag",
         defaultName: "Documentation",
diff --git a/packages/components/tsconfig.json b/packages/components/tsconfig.json
index 2d5dca81..efc119ae 100644
--- a/packages/components/tsconfig.json
+++ b/packages/components/tsconfig.json
@@ -23,7 +23,8 @@
         "esModuleInterop": true
     },
     "include": [
-        "lib"
+        "lib",
+        "index.ts"
     ],
     "references": [
         {
diff --git a/packages/components/vite.config.ts b/packages/components/vite.config.ts
index 851fdbc1..f5be834b 100644
--- a/packages/components/vite.config.ts
+++ b/packages/components/vite.config.ts
@@ -1,5 +1,5 @@
-import {defineConfig} from "vite";
-import {resolve} from "path";
+import { defineConfig } from "vite";
+import { resolve } from "path";
 import react from "@vitejs/plugin-react";
 import dts from "vite-plugin-dts";
 
@@ -9,13 +9,12 @@ export default defineConfig({
         react(),
         dts({
             insertTypesEntry: true,
-            rollupTypes: true,
             exclude: ["**/*.stories.tsx", "**/*.test.tsx"],
         }),
     ],
     build: {
         lib: {
-            entry: resolve(__dirname, "lib/index.ts"),
+            entry: resolve(__dirname, "index.ts"),
             name: "vertex-components",
             formats: ["es", "umd"],
             fileName: (format) => `vertex-components.${format}.js`,