diff --git a/query_data.py b/query_data.py
index c0028317f..167146587 100644
--- a/query_data.py
+++ b/query_data.py
@@ -1,5 +1,5 @@
 """Create a ChatVectorDBChain for question/answering."""
-from langchain.callbacks.base import AsyncCallbackManager
+from langchain.callbacks.manager import AsyncCallbackManager
 from langchain.callbacks.tracers import LangChainTracer
 from langchain.chains import ChatVectorDBChain
 from langchain.chains.chat_vector_db.prompts import (CONDENSE_QUESTION_PROMPT,
diff --git a/requirements.txt b/requirements.txt
index 1b7831d96..fdf772eb1 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -11,3 +11,4 @@ faiss-cpu
 bs4
 unstructured
 libmagic
+tiktoken