diff --git a/dsLightRag/Test/T1_WriteToPg.py b/dsLightRag/Test/T1_WriteToPg.py
index 458dbcdd..2598a18c 100644
--- a/dsLightRag/Test/T1_WriteToPg.py
+++ b/dsLightRag/Test/T1_WriteToPg.py
@@ -39,7 +39,7 @@ os.environ["POSTGRES_PASSWORD"] = "postgres"
 os.environ["POSTGRES_DATABASE"] = "rag"
 
 
-async def initialize_rag():
+async def initialize_pg_rag():
     rag = LightRAG(
         working_dir=WORKING_DIR,
         llm_model_func=llm_model_func,
@@ -66,7 +66,7 @@ async def initialize_rag():
 
 
 async def main():
-    rag = await initialize_rag()
+    rag = await initialize_pg_rag()
 
     with open(f"../Txt/sushi.txt", "r", encoding="utf-8") as f:
         await rag.ainsert(f.read())
diff --git a/dsLightRag/Test/T2_ReadFromPg.py b/dsLightRag/Test/T2_ReadFromPg.py
index 31806422..377d0b14 100644
--- a/dsLightRag/Test/T2_ReadFromPg.py
+++ b/dsLightRag/Test/T2_ReadFromPg.py
@@ -40,7 +40,7 @@ os.environ["POSTGRES_PASSWORD"] = "postgres"
 os.environ["POSTGRES_DATABASE"] = "rag"
 
 
-async def initialize_rag():
+async def initialize_pg_rag():
     rag = LightRAG(
         working_dir=WORKING_DIR,
         llm_model_func=llm_model_func,
@@ -67,7 +67,7 @@ async def initialize_rag():
 
 async def main():
     try:
-        rag = await initialize_rag()
+        rag = await initialize_pg_rag()
         resp = await rag.aquery(
             "What are the top themes in this story",
             param=QueryParam(mode="hybrid", stream=True),