diff --git a/templates/rag-lancedb/rag_lancedb/chain.py b/templates/rag-lancedb/rag_lancedb/chain.py
index 9ba07e3664f..856d65f0f22 100644
--- a/templates/rag-lancedb/rag_lancedb/chain.py
+++ b/templates/rag-lancedb/rag_lancedb/chain.py
@@ -19,10 +19,8 @@ text_splitter = RecursiveCharacterTextSplitter(chunk_size=500, chunk_overlap=0)
 all_splits = text_splitter.split_documents(data)
 
 # Add to vectorDB
-vectorstore = LanceDB.from_documents(documents=all_splits, 
-                                    collection_name="rag-chroma",
-                                    embedding=OpenAIEmbeddings(),
-                                    )
+vectorstore = LanceDB.from_documents(documents=all_splits,
+                             embedding=OpenAIEmbeddings())
 retriever = vectorstore.as_retriever()
 """