diff --git a/packages/payload-ai/package.json b/packages/payload-ai/package.json
index 87cb077..871f7c8 100644
--- a/packages/payload-ai/package.json
+++ b/packages/payload-ai/package.json
@@ -11,12 +11,13 @@
   "types": "./dist/index.d.ts",
   "exports": {
     ".": "./dist/index.js",
-    "./components/*": "./dist/components/*",
-    "./actions/*": "./dist/actions/*",
-    "./globals/*": "./dist/globals/*",
-    "./endpoints/*": "./dist/endpoints/*",
-    "./utils/*": "./dist/utils/*",
-    "./tools/*": "./dist/tools/*"
+    "./components/FieldGenerators/*": "./dist/components/FieldGenerators/*.js",
+    "./components/*": "./dist/components/*.js",
+    "./actions/*": "./dist/actions/*.js",
+    "./globals/*": "./dist/globals/*.js",
+    "./endpoints/*": "./dist/endpoints/*.js",
+    "./utils/*": "./dist/utils/*.js",
+    "./tools/*": "./dist/tools/*.js"
   },
   "peerDependencies": {
     "@payloadcms/next": ">=3.0.0",
diff --git a/packages/payload-ai/src/chatPlugin.ts b/packages/payload-ai/src/chatPlugin.ts
index 355e161..71cb6ef 100644
--- a/packages/payload-ai/src/chatPlugin.ts
+++ b/packages/payload-ai/src/chatPlugin.ts
@@ -88,7 +88,7 @@ export const payloadChatPlugin =
           ...(config.admin?.components || {}),
           providers: [
             ...(config.admin?.components?.providers || []),
-            "@mintel/payload-ai/components/ChatWindow#ChatWindowProvider",
+            "@mintel/payload-ai/components/ChatWindow/index#ChatWindowProvider",
           ],
         },
       };
diff --git a/packages/payload-ai/src/components/ChatWindow/index.tsx b/packages/payload-ai/src/components/ChatWindow/index.tsx
index 29af98a..a95c031 100644
--- a/packages/payload-ai/src/components/ChatWindow/index.tsx
+++ b/packages/payload-ai/src/components/ChatWindow/index.tsx
@@ -2,7 +2,6 @@
 
 import React, { useState, useEffect } from "react";
 import { useChat } from "@ai-sdk/react";
-import "./ChatWindow.scss";
 
 export const ChatWindowProvider: React.FC<{ children: React.ReactNode }> = ({
   children,