diff --git a/scripts/rag_build_query.py b/scripts/rag_build_query.py
index 17a0947..f59f6b8 100644
--- a/scripts/rag_build_query.py
+++ b/scripts/rag_build_query.py
@@ -1,17 +1,16 @@
 import os
 import faiss
 from typing import List
-from llama_index import (
-    SimpleDirectoryReader,
-    VectorStoreIndex,
-    ServiceContext,
-    PromptTemplate,
-)
+from llama_index.readers.file import SimpleDirectoryReader
+from llama_index.indices.vector_store import VectorStoreIndex
+from llama_index.service_context import ServiceContext
+from llama_index.prompts import PromptTemplate
 from llama_index.embeddings.huggingface import HuggingFaceEmbedding
 from llama_index.vector_stores.faiss import FaissVectorStore
 from app.core.config import settings
 from scripts.permissions import get_user_allowed_indexes
 
+
 USER_INDEX_PATH = "index_data"
 USER_DOC_PATH = "docs"