diff --git a/docs/docs/modules/callbacks/index.mdx b/docs/docs/modules/callbacks/index.mdx
index 9d0a1c78c74..3a3ba08cfb8 100644
--- a/docs/docs/modules/callbacks/index.mdx
+++ b/docs/docs/modules/callbacks/index.mdx
@@ -95,42 +95,40 @@ prompt = PromptTemplate.from_template("1 + {number} = ")
 
 # Constructor callback: First, let's explicitly set the StdOutCallbackHandler when initializing our chain
 chain = LLMChain(llm=llm, prompt=prompt, callbacks=[handler])
-chain.run(number=2)
+chain.invoke({"number":2})
 
 # Use verbose flag: Then, let's use the `verbose` flag to achieve the same result
 chain = LLMChain(llm=llm, prompt=prompt, verbose=True)
-chain.run(number=2)
+chain.invoke({"number":2})
 
 # Request callbacks: Finally, let's use the request `callbacks` to achieve the same result
 chain = LLMChain(llm=llm, prompt=prompt)
-chain.run(number=2, callbacks=[handler])
+chain.invoke({"number":2}, {"callbacks":[handler]})
+
 ```
 
 <CodeOutputBlock lang="python">
 
 ```
-    > Entering new LLMChain chain...
-    Prompt after formatting:
-    1 + 2 =
+> Entering new LLMChain chain...
+Prompt after formatting:
+1 + 2 = 
 
-    > Finished chain.
+> Finished chain.
 
 
-    > Entering new LLMChain chain...
-    Prompt after formatting:
-    1 + 2 =
+> Entering new LLMChain chain...
+Prompt after formatting:
+1 + 2 = 
 
-    > Finished chain.
+> Finished chain.
 
 
-    > Entering new LLMChain chain...
-    Prompt after formatting:
-    1 + 2 =
+> Entering new LLMChain chain...
+Prompt after formatting:
+1 + 2 = 
 
-    > Finished chain.
-
-
-    '\n\n3'
+> Finished chain.
 ```
 
 </CodeOutputBlock>