diff --git a/lightrag/prompt.py b/lightrag/prompt.py
index 8ea38abc..75585553 100644
--- a/lightrag/prompt.py
+++ b/lightrag/prompt.py
@@ -236,12 +236,13 @@ Consider the conversation history if provided to maintain conversational flow an
   - Target format and length: {response_type}
 
 4. **Citation Format:**
-  - All citations should be consolidated in the References section at the end of the response under a `### References` heading. Do not include citation information within the main body of the answer.
+  - All citations should be consolidated in the References section at the end of the response. Do not include citation information within the main body of the answer.
+  - The References section should be under a `### References` heading.
   - Output the citation in the following formats:
     - For a Knowledge Graph Entity: `[KG] <entity_name>`
     - For a Knowledge Graph Relationship: `[KG] <entity1_name> ~ <entity2_name>`
     - For a Document Chunk: `[DC] <file_path_or_document_name>`
-  - Output each citation on a single new line.
+  - List each citation on a separate new line.
   - Provide a maximum of 5 unique and most relevant references. Each entity, relationship, or document name must appear only once in the "References" section.
 
 ---Source Data---
@@ -276,9 +277,10 @@ Consider the conversation history if provided to maintain conversational flow an
   - Target format and length: {response_type}
 
 4. **Citation Format:**
-  - All citations should be consolidated in the References section at the end of the response under a `### References` heading. Do not include citation information within the main body of the answer.
+  - All citations should be consolidated in the References section at the end of the response. Do not include citation information within the main body of the answer.
+  - The References section should be under a `### References` heading.
   - Output the citation in the following format: `[DC] <file_path_or_document_name>`
-  - Output each citation on a single new line.
+  - List each citation on a separate new line.
   - Provide a maximum of 5 unique and most relevant references. Each entity, relationship, or document name must appear only once in the "References" section.
 
 ---Source Data---