diff --git a/lightrag/kg/postgres_impl.py b/lightrag/kg/postgres_impl.py
index cb636d7f..377d5979 100644
--- a/lightrag/kg/postgres_impl.py
+++ b/lightrag/kg/postgres_impl.py
@@ -587,7 +587,7 @@ class PGGraphStorage(BaseGraphStorage):
         print("no preloading of graph with AGE in production")
 
     def __post_init__(self):
-        self.graph_name = os.environ.get("AGE_GRAPH_NAME", "lightrag")
+        self.graph_name = self.namespace or os.environ.get("AGE_GRAPH_NAME", "lightrag")
         self._node_embed_algorithms = {
             "node2vec": self._node2vec_embed,
         }