diff --git a/backend/chainlit/langchain/callbacks.py b/backend/chainlit/langchain/callbacks.py
index f841c0c56d..ef0b9b7182 100644
--- a/backend/chainlit/langchain/callbacks.py
+++ b/backend/chainlit/langchain/callbacks.py
@@ -3,11 +3,11 @@
 from uuid import UUID
 
 import pydantic
-from langchain.callbacks.tracers.schemas import Run
-from langchain.load.dump import dumps
-from langchain.schema import BaseMessage
+from langchain_core.load import dumps
+from langchain_core.messages import BaseMessage
 from langchain_core.outputs import ChatGenerationChunk, GenerationChunk
 from langchain_core.tracers.base import AsyncBaseTracer
+from langchain_core.tracers.schemas import Run
 from literalai import ChatGeneration, CompletionGeneration, GenerationMessage
 from literalai.observability.step import TrueStepType