diff --git a/docs/my-website/docs/observability/langfuse_integration.md b/docs/my-website/docs/observability/langfuse_integration.md
index 102d9d0ba..3d0e17271 100644
--- a/docs/my-website/docs/observability/langfuse_integration.md
+++ b/docs/my-website/docs/observability/langfuse_integration.md
@@ -138,6 +138,7 @@ Pass `trace_user_id`, `session_id` in model_kwargs
 import os
 from langchain.chat_models import ChatLiteLLM
 from langchain.schema import HumanMessage
+import litellm
 
 # from https://cloud.langfuse.com/
 os.environ["LANGFUSE_PUBLIC_KEY"] = ""