diff --git a/lightrag/kg/postgres_impl.py b/lightrag/kg/postgres_impl.py
index c067a0d0..346e0496 100644
--- a/lightrag/kg/postgres_impl.py
+++ b/lightrag/kg/postgres_impl.py
@@ -1819,10 +1819,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 (