diff --git a/src/oss/javascript/migrate/langchain-v1.mdx b/src/oss/javascript/migrate/langchain-v1.mdx
index 848b1ab0c..51c3c48f2 100644
--- a/src/oss/javascript/migrate/langchain-v1.mdx
+++ b/src/oss/javascript/migrate/langchain-v1.mdx
@@ -111,11 +111,11 @@ import { createAgent, dynamicSystemPromptMiddleware } from "langchain";
 import * as z from "zod";
 
 const contextSchema = z.object({
-  userRole: z.enum(["expert", "beginner"]).default("user"),
+  userRole: z.enum(["expert", "beginner"]),
 });
 
-const userRolePrompt = dynamicSystemPromptMiddleware((request) => { // [!code highlight]
-  const userRole = request.runtime.context.userRole;
+const userRolePrompt = dynamicSystemPromptMiddleware((state, runtime) => { // [!code highlight]
+  const userRole = runtime.context.userRole;
   const basePrompt = "You are a helpful assistant.";
 
   if (userRole === "expert") {
@@ -135,6 +135,7 @@ const agent = createAgent({
 
 await agent.invoke({
   messages: [new HumanMessage("Explain async programming")],
+},{
   context: {
     userRole: "expert",
   },
@@ -145,7 +146,7 @@ await agent.invoke({
 import { createReactAgent } from "@langchain/langgraph/prebuilts";
 
 const contextSchema = z.object({
-  userRole: z.enum(["expert", "beginner"]).default("user"),
+  userRole: z.enum(["expert", "beginner"]),
 });
 
 const agent = createReactAgent({
@@ -167,7 +168,8 @@ const agent = createReactAgent({
 
 // Use with context
 await agent.invoke({
-  messages: [new HumanMessage("Explain async programming")],
+  messages: [new HumanMessage("Explain async programming")]
+}, {
   context: { userRole: "expert" },
 });
 ```