diff --git a/a2rchi/chains/chain.py b/a2rchi/chains/chain.py
index 6103de8..9c2ba94 100644
--- a/a2rchi/chains/chain.py
+++ b/a2rchi/chains/chain.py
@@ -1,7 +1,7 @@
 from a2rchi.chains.base import BaseSubMITChain as BaseChain
 
 from chromadb.config import Settings
-from langchain_chroma import Chroma
+from langchain_chroma.vectorstores import Chroma
 
 import chromadb
 import time
diff --git a/a2rchi/utils/data_manager.py b/a2rchi/utils/data_manager.py
index 535e415..a2680e4 100644
--- a/a2rchi/utils/data_manager.py
+++ b/a2rchi/utils/data_manager.py
@@ -5,7 +5,7 @@
 from langchain_community.document_loaders import PyPDFLoader
 from langchain_community.document_loaders import BSHTMLLoader
 from langchain_openai import OpenAIEmbeddings
-from langchain_chroma import Chroma
+from langchain_chroma.vectorstores import Chroma
 from langchain_text_splitters.character import CharacterTextSplitter
 
 import chromadb