diff --git a/README.md b/README.md
index 269e8249..02ecbe4e 100644
--- a/README.md
+++ b/README.md
@@ -321,7 +321,7 @@ WORKING_DIR = "./local_neo4jWorkDir"
 rag = LightRAG(
     working_dir=WORKING_DIR,
     llm_model_func=gpt_4o_mini_complete,  # Use gpt_4o_mini_complete LLM model
-    kg="Neo4JStorage", #<-----------override KG default
+    graph_storage="Neo4JStorage", #<-----------override KG default
     log_level="DEBUG"  #<-----------override log_level default
 )
 ```
diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py
index 6511e61b..2b2d4820 100644
--- a/lightrag/lightrag.py
+++ b/lightrag/lightrag.py
@@ -186,7 +186,9 @@ class LightRAG:
             embedding_func=self.embedding_func,
         )
         self.chunk_entity_relation_graph = self.graph_storage_cls(
-            namespace="chunk_entity_relation", global_config=asdict(self)
+            namespace="chunk_entity_relation",
+            global_config=asdict(self),
+            embedding_func=self.embedding_func,
         )
         ####
         # add embedding func by walter over
diff --git a/test_neo4j.py b/test_neo4j.py
index 822cd7bc..0048fc17 100644
--- a/test_neo4j.py
+++ b/test_neo4j.py
@@ -17,7 +17,7 @@ if not os.path.exists(WORKING_DIR):
 rag = LightRAG(
     working_dir=WORKING_DIR,
     llm_model_func=gpt_4o_mini_complete,  # Use gpt_4o_mini_complete LLM model
-    kg="Neo4JStorage",
+    graph_storage="Neo4JStorage",
     log_level="INFO",
     # llm_model_func=gpt_4o_complete  # Optionally, use a stronger model
 )