diff --git a/lightrag/prompt.py b/lightrag/prompt.py
index e89f4b4b..0cd8660b 100644
--- a/lightrag/prompt.py
+++ b/lightrag/prompt.py
@@ -26,8 +26,8 @@ For a given text and a list of entity types, extract all entities and their rela
   - relationship_description: Explain the nature of the relationship between the source and target entities, providing a clear rationale for their connection
 4. Format each relationship as: (relationship{tuple_delimiter}<source_entity>{tuple_delimiter}<target_entity>{tuple_delimiter}<relationship_keywords>{tuple_delimiter}<relationship_description>)
 5. Use `{tuple_delimiter}` as field delimiter. Use `{record_delimiter}` as the entity or relation list delimiter.
-6. Return identified entities and relationships in {language}.
-7. Output `{completion_delimiter}` when all the entities and relationships are extracted.
+6. Output `{completion_delimiter}` when all the entities and relationships are extracted.
+7. Ensure the output language is {language}.
 
 ---Quality Guidelines---
 - Only extract entities and relationships that are clearly defined and meaningful in the context