diff --git a/docs/my-website/docs/image_generation.md b/docs/my-website/docs/image_generation.md
index 25327df79..7d436148f 100644
--- a/docs/my-website/docs/image_generation.md
+++ b/docs/my-website/docs/image_generation.md
@@ -14,7 +14,7 @@ response = image_generation(prompt="A cute baby sea otter", model="dall-e-3")
 print(f"response: {response}")
 ```
 
-### Input Params for `litellm.embedding()`
+### Input Params for `litellm.image_generation()`
 ### Required Fields
 
 - `prompt`: *string* - A text description of the desired image(s).