diff --git a/app/rag_chain.py b/app/rag_chain.py
index 8d96953..3ae09db 100644
--- a/app/rag_chain.py
+++ b/app/rag_chain.py
@@ -2,6 +2,7 @@
 import os
 import weaviate
 from weaviate import Client
+from weaviate import WeaviateClient
 from weaviate.connect import ConnectionParams
 from weaviate.auth import AuthApiKey
 from weaviate.classes.init import Auth