diff --git a/packages/kilo-ui/src/stories/session-review.stories.tsx b/packages/kilo-ui/src/stories/session-review.stories.tsx
index f2f201f007b..f01390f7d53 100644
--- a/packages/kilo-ui/src/stories/session-review.stories.tsx
+++ b/packages/kilo-ui/src/stories/session-review.stories.tsx
@@ -1,18 +1,18 @@
 /** @jsxImportSource solid-js */
 import type { Meta, StoryObj } from "storybook-solidjs-vite"
 import { SessionReview } from "@opencode-ai/ui/session-review"
-import { DiffComponentProvider } from "@kilocode/kilo-ui/context/diff"
-import { Diff } from "@kilocode/kilo-ui/diff"
+import { FileComponentProvider } from "@kilocode/kilo-ui/context/file"
+import { File } from "@kilocode/kilo-ui/file"
 
 const meta: Meta = {
   title: "Components/SessionReview",
   decorators: [
     (Story) => (
-      <DiffComponentProvider component={Diff}>
+      <FileComponentProvider component={File}>
         <div style={{ width: "800px", "min-height": "300px" }}>
           <Story />
         </div>
-      </DiffComponentProvider>
+      </FileComponentProvider>
     ),
   ],
   parameters: { layout: "padded" },
diff --git a/packages/kilo-ui/src/stories/session-turn.stories.tsx b/packages/kilo-ui/src/stories/session-turn.stories.tsx
index 9df74cc0608..92ffa3ac65a 100644
--- a/packages/kilo-ui/src/stories/session-turn.stories.tsx
+++ b/packages/kilo-ui/src/stories/session-turn.stories.tsx
@@ -2,10 +2,10 @@
 import type { Meta, StoryObj } from "storybook-solidjs-vite"
 import { SessionTurn } from "@opencode-ai/ui/session-turn"
 import { DataProvider } from "@opencode-ai/ui/context/data"
-import { DiffComponentProvider } from "@kilocode/kilo-ui/context/diff"
+import { FileComponentProvider } from "@kilocode/kilo-ui/context/file"
 import { DialogProvider } from "@opencode-ai/ui/context/dialog"
 import { MarkedProvider } from "@opencode-ai/ui/context/marked"
-import { Diff } from "@kilocode/kilo-ui/diff"
+import { File } from "@kilocode/kilo-ui/file"
 import type { UserMessage, AssistantMessage, TextPart, ToolPart } from "@kilocode/sdk/v2"
 
 const SESSION_ID = "session-turn-story-001"
@@ -151,11 +151,11 @@ const workingData = {
 function Providers(props: { data: any; children: any }) {
   return (
     <DataProvider data={props.data} directory="/project">
-      <DiffComponentProvider component={Diff}>
+      <FileComponentProvider component={File}>
         <DialogProvider>
           <MarkedProvider>{props.children}</MarkedProvider>
         </DialogProvider>
-      </DiffComponentProvider>
+      </FileComponentProvider>
     </DataProvider>
   )
 }
@@ -172,7 +172,7 @@ export const Default: Story = {
   render: () => (
     <Providers data={baseData}>
       <div style={{ width: "700px" }}>
-        <SessionTurn sessionID={SESSION_ID} messageID={USER_MSG_ID} lastUserMessageID={USER_MSG_ID} />
+        <SessionTurn sessionID={SESSION_ID} messageID={USER_MSG_ID} />
       </div>
     </Providers>
   ),
@@ -182,7 +182,7 @@ export const Working: Story = {
   render: () => (
     <Providers data={workingData}>
       <div style={{ width: "700px" }}>
-        <SessionTurn sessionID={SESSION_ID_2} messageID={USER_MSG_ID_2} lastUserMessageID={USER_MSG_ID_2} />
+        <SessionTurn sessionID={SESSION_ID_2} messageID={USER_MSG_ID_2} />
       </div>
     </Providers>
   ),
@@ -192,7 +192,7 @@ export const WithStepsExpanded: Story = {
   render: () => (
     <Providers data={baseData}>
       <div style={{ width: "700px" }}>
-        <SessionTurn sessionID={SESSION_ID} messageID={USER_MSG_ID} lastUserMessageID={USER_MSG_ID} stepsExpanded />
+        <SessionTurn sessionID={SESSION_ID} messageID={USER_MSG_ID} stepsExpanded />
       </div>
     </Providers>
   ),