diff --git a/customer-graph/.env.example b/customer-graph/.env.example
index d916980..390d984 100644
--- a/customer-graph/.env.example
+++ b/customer-graph/.env.example
@@ -2,6 +2,7 @@
 NEO4J_URI=<YOUR_NEO4J_URI>
 NEO4J_USERNAME=<YOUR_NEO4J_USERNAME>
 NEO4J_PASSWORD=<YOUR_NEO4J_PASSWORD>
+NEO4J_DATABASE=<NEO4J_DATABASE>
 
 #OpenAI
 OPENAI_API_KEY=<YOUR_OPENAI_API_KEY>
\ No newline at end of file
diff --git a/customer-graph/ingest_post_processing.py b/customer-graph/ingest_post_processing.py
index e8c574b..cede694 100644
--- a/customer-graph/ingest_post_processing.py
+++ b/customer-graph/ingest_post_processing.py
@@ -7,6 +7,7 @@
 NEO4J_URI=os.getenv("NEO4J_URI")
 NEO4J_USERNAME=os.getenv("NEO4J_USERNAME")
 NEO4J_PASSWORD=os.getenv("NEO4J_PASSWORD")
+NEO4J_DATABASE=os.getenv("NEO4J_DATABASE")
 
 
 
@@ -30,7 +31,7 @@
 
 # create text embeddings for products
 print("Creating Product Text Embeddings")
-with driver.session(database="neo4j") as session:
+with driver.session(database=NEO4J_DATABASE) as session:
     session.run('''
     MATCH (n:Product) WHERE size(n.description) <> 0
     WITH collect(n) AS nodes, toInteger(rand()*$numberOfBatches) AS partition