diff --git a/lightrag/operate.py b/lightrag/operate.py
index 89f9a777..d5dfc793 100644
--- a/lightrag/operate.py
+++ b/lightrag/operate.py
@@ -362,7 +362,7 @@ async def extract_entities(
         tuple_delimiter=PROMPTS["DEFAULT_TUPLE_DELIMITER"],
         record_delimiter=PROMPTS["DEFAULT_RECORD_DELIMITER"],
         completion_delimiter=PROMPTS["DEFAULT_COMPLETION_DELIMITER"],
-        entity_types=",".join(entity_types),
+        entity_types=", ".join(entity_types),
         language=language,
     )
     # add example's format
diff --git a/lightrag/prompt.py b/lightrag/prompt.py
index c6cbf7db..d22244c9 100644
--- a/lightrag/prompt.py
+++ b/lightrag/prompt.py
@@ -47,8 +47,9 @@ Format the content-level key words as ("content_keywords"{tuple_delimiter}<high_
 #############################
 ---Real Data---
 ######################
-Entity_types: {entity_types}
-Text: {input_text}
+Entity_types: [{entity_types}]
+Text:
+{input_text}
 ######################
 Output:"""