diff --git a/dataherald/context_store/__init__.py b/dataherald/context_store/__init__.py
index 2ae52dbc..e61f292d 100644
--- a/dataherald/context_store/__init__.py
+++ b/dataherald/context_store/__init__.py
@@ -17,9 +17,7 @@ class ContextStore(Component, ABC):
     def __init__(self, system: System):
         self.system = system
         self.db = self.system.instance(DB)
-        self.golden_sql_collection = os.environ.get(
-            "GOLDEN_SQL_COLLECTION", "dataherald-staging"
-        )
+        self.golden_sql_collection = os.environ.get("GOLDEN_SQL_COLLECTION", "ai-stage")
         self.vector_store = self.system.instance(VectorStore)
 
     @abstractmethod
diff --git a/dataherald/scripts/delete_and_populate_golden_records.py b/dataherald/scripts/delete_and_populate_golden_records.py
index cc6bad8e..28d65874 100644
--- a/dataherald/scripts/delete_and_populate_golden_records.py
+++ b/dataherald/scripts/delete_and_populate_golden_records.py
@@ -11,9 +11,7 @@
     system = System(settings)
     system.start()
     storage = system.instance(DB)
-    golden_sql_collection = os.environ.get(
-        "GOLDEN_SQL_COLLECTION", "dataherald-staging"
-    )
+    golden_sql_collection = os.environ.get("GOLDEN_SQL_COLLECTION", "ai-stage")
 
     golden_sqls = storage.find_all("golden_sqls")
     vector_store = system.instance(VectorStore)
diff --git a/dataherald/scripts/migrate_v001_to_v002.py b/dataherald/scripts/migrate_v001_to_v002.py
index a45ce309..143b3c38 100644
--- a/dataherald/scripts/migrate_v001_to_v002.py
+++ b/dataherald/scripts/migrate_v001_to_v002.py
@@ -35,9 +35,7 @@ def add_db_connection_id(collection_name: str, storage) -> None:
     add_db_connection_id("golden_records", storage)
     add_db_connection_id("nl_question", storage)
     # Refresh vector stores
-    golden_record_collection = os.environ.get(
-        "GOLDEN_SQL_COLLECTION", "dataherald-staging"
-    )
+    golden_record_collection = os.environ.get("GOLDEN_SQL_COLLECTION", "ai-stage")
     vector_store = system.instance(VectorStore)
     try:
         vector_store.delete_collection(golden_record_collection)
diff --git a/dataherald/scripts/migrate_v006_to_v100.py b/dataherald/scripts/migrate_v006_to_v100.py
index 6f15e115..4cf330d9 100644
--- a/dataherald/scripts/migrate_v006_to_v100.py
+++ b/dataherald/scripts/migrate_v006_to_v100.py
@@ -30,9 +30,7 @@ def update_object_id_fields(field_name: str, collection_name: str):
     system.start()
     storage = system.instance(DB)
     # Refresh vector stores
-    golden_sql_collection = os.environ.get(
-        "GOLDEN_RECORD_COLLECTION", "dataherald-staging"
-    )
+    golden_sql_collection = os.environ.get("GOLDEN_RECORD_COLLECTION", "ai-stage")
     vector_store = system.instance(VectorStore)
 
     golden_records = storage.find_all("golden_records")