diff --git a/lightrag/kg/postgres_impl.py b/lightrag/kg/postgres_impl.py
index ca5da8f1..bd133ab0 100644
--- a/lightrag/kg/postgres_impl.py
+++ b/lightrag/kg/postgres_impl.py
@@ -2100,10 +2100,10 @@ SQL_TEMPLATES = {
                 FROM LIGHTRAG_VDB_ENTITY e
                 JOIN relevant_chunks c ON c.chunk_id = ANY(e.chunk_ids)
                 WHERE e.workspace=$1
-            )
-        WHERE distance>$2
-        ORDER BY distance DESC
-        LIMIT $3
+            ) as chunk_distances
+            WHERE distance>$2
+            ORDER BY distance DESC
+            LIMIT $3
     """,
     "chunks": """
         WITH relevant_chunks AS (