diff --git a/docs/my-website/docs/providers/deepinfra.md b/docs/my-website/docs/providers/deepinfra.md
index d8466848db..dad7f5d2cd 100644
--- a/docs/my-website/docs/providers/deepinfra.md
+++ b/docs/my-website/docs/providers/deepinfra.md
@@ -2,7 +2,7 @@
 https://deepinfra.com/
 ## Sample Usage
 ```python
-import litellm
+from litellm import completion
 import os
 
 os.environ['DEEPINFRA_API_KEY'] = ""
@@ -14,7 +14,7 @@ response = completion(
 
 ## Sample Usage - Streaming
 ```python
-import litellm
+from litellm import completion
 import os
 
 os.environ['DEEPINFRA_API_KEY'] = ""