diff --git a/apps/www/pages/docs/advanced/Branching.mdx b/apps/www/pages/docs/advanced/Branching.mdx
index eac9c65cc..d8687f1f4 100644
--- a/apps/www/pages/docs/advanced/Branching.mdx
+++ b/apps/www/pages/docs/advanced/Branching.mdx
@@ -16,7 +16,7 @@ Branches are automatically tracked by assistant-ui by observing changes to the `
 You can show a branch picker by using `BranchPickerPrimitive`.
 
 ```tsx {1, 8, 15-30}
-import { BranchPickerPrimitive } from "assistant-ui";
+import { BranchPickerPrimitive } from "@assistant-ui/react";
 
 
 const Message = () => {
diff --git a/apps/www/pages/docs/advanced/Editing.mdx b/apps/www/pages/docs/advanced/Editing.mdx
index 4f4b9d23b..a8f72ea40 100644
--- a/apps/www/pages/docs/advanced/Editing.mdx
+++ b/apps/www/pages/docs/advanced/Editing.mdx
@@ -8,7 +8,7 @@ description: Give the user the ability to edit their message.
 You can show an editor interface by using `ComposerPrimitive`.
 
 ```tsx {1,11,25,31-43}
-import { ComposerPrimitive } from "assistant-ui";
+import { ComposerPrimitive } from "@assistant-ui/react";
 ...
 
 const Thread = () => {
diff --git a/apps/www/pages/reference/primitives/ActionBar.mdx b/apps/www/pages/reference/primitives/ActionBar.mdx
index 2a4de8665..fa28d483b 100644
--- a/apps/www/pages/reference/primitives/ActionBar.mdx
+++ b/apps/www/pages/reference/primitives/ActionBar.mdx
@@ -9,7 +9,7 @@ import { Code } from "@radix-ui/themes";
 ## Anatomy
 
 ```tsx
-import { ActionBarPrimitive } from "assistant-ui";
+import { ActionBarPrimitive } from "@assistant-ui/react";
 
 const UserMessageBar = () => (
   <ActionBarPrimitive.Root>
diff --git a/apps/www/pages/reference/primitives/BranchPicker.mdx b/apps/www/pages/reference/primitives/BranchPicker.mdx
index c9a650b4e..cdd16e7a0 100644
--- a/apps/www/pages/reference/primitives/BranchPicker.mdx
+++ b/apps/www/pages/reference/primitives/BranchPicker.mdx
@@ -8,7 +8,7 @@ import { PropsTable } from "@/components/docs";
 ## Anatomy
 
 ```tsx
-import { BranchPickerPrimitive } from "assistant-ui";
+import { BranchPickerPrimitive } from "@assistant-ui/react";
 
 const BranchPicker = () => (
   <BranchPickerPrimitive.Root>
diff --git a/apps/www/pages/reference/primitives/Composer.mdx b/apps/www/pages/reference/primitives/Composer.mdx
index 22f6edb4a..7e46bfa99 100644
--- a/apps/www/pages/reference/primitives/Composer.mdx
+++ b/apps/www/pages/reference/primitives/Composer.mdx
@@ -15,7 +15,7 @@ import { Code } from "@radix-ui/themes";
 ## Anatomy
 
 ```tsx
-import { ComposerPrimitive } from "assistant-ui";
+import { ComposerPrimitive } from "@assistant-ui/react";
 
 // creating a new message
 const Composer = () => (
diff --git a/apps/www/pages/reference/primitives/ContentPart.mdx b/apps/www/pages/reference/primitives/ContentPart.mdx
index 7ccf4cba3..04c473a4c 100644
--- a/apps/www/pages/reference/primitives/ContentPart.mdx
+++ b/apps/www/pages/reference/primitives/ContentPart.mdx
@@ -6,7 +6,7 @@ description: A part of a message's content. Content parts may be text, image, to
 ## Anatomy
 
 ```tsx
-import { ContentPartPrimitive } from "assistant-ui";
+import { ContentPartPrimitive } from "@assistant-ui/react";
 
 const TextContentPart = () => {
   return (
diff --git a/apps/www/pages/reference/primitives/Message.mdx b/apps/www/pages/reference/primitives/Message.mdx
index d6f6d9f00..2efbdf8cf 100644
--- a/apps/www/pages/reference/primitives/Message.mdx
+++ b/apps/www/pages/reference/primitives/Message.mdx
@@ -8,7 +8,7 @@ import { PropsTable } from "@/components/docs";
 ## Anatomy
 
 ```tsx
-import { MessagePrimitive } from "assistant-ui";
+import { MessagePrimitive } from "@assistant-ui/react";
 
 const UserMessage = () => (
   <MessagePrimitive.Root>
diff --git a/apps/www/pages/reference/primitives/Thread.mdx b/apps/www/pages/reference/primitives/Thread.mdx
index 24ba9db3e..7a0fcf34b 100644
--- a/apps/www/pages/reference/primitives/Thread.mdx
+++ b/apps/www/pages/reference/primitives/Thread.mdx
@@ -8,7 +8,7 @@ import { PropsTable } from "@/components/docs";
 ## Anatomy
 
 ```tsx
-import { ThreadPrimitive } from "assistant-ui";
+import { ThreadPrimitive } from "@assistant-ui/react";
 
 const Thread = () => (
   <ThreadPrimitive.Root>