diff --git a/lightrag/kg/postgres_impl.py b/lightrag/kg/postgres_impl.py
index a2e9be38..d2630659 100644
--- a/lightrag/kg/postgres_impl.py
+++ b/lightrag/kg/postgres_impl.py
@@ -1766,7 +1766,7 @@ SQL_TEMPLATES = {
                 FROM LIGHTRAG_DOC_CHUNKS
                 where workspace=$1
                 AND id IN (SELECT chunk_id FROM relevant_chunks)
-            )
+            ) as chunk_distances
             WHERE distance>$2
             ORDER BY distance DESC
             LIMIT $3
diff --git a/lightrag/operate.py b/lightrag/operate.py
index 1815f308..d062ae73 100644
--- a/lightrag/operate.py
+++ b/lightrag/operate.py
@@ -404,7 +404,7 @@ async def extract_entities(
         language=language,
     )
 
-    continue_prompt = PROMPTS["entity_continue_extraction"]
+    continue_prompt = PROMPTS["entity_continue_extraction"].format(**context_base)
     if_loop_prompt = PROMPTS["entity_if_loop_extraction"]
 
     processed_chunks = 0