diff --git a/scripts/custom.config.sh b/scripts/custom.config.sh
index fc96e1983cbd30d045e3996a02f3aeb2f5caea72..a00c5b9a5f9b2d9d894cb74f76e487c5b1e4cf7b 100644
--- a/scripts/custom.config.sh
+++ b/scripts/custom.config.sh
@@ -8,3 +8,7 @@ export SERVICE_MAP=(
 export DATA_HOST=${DATA_HOST:-cwbpub.cr.usgs.gov};
 export DATA_PORT=${DATA_PORT:-2060};
 export DATA_TYPE=${DATA_TYPE:-edge};
+
+if [[ $TARGET_HOSTNAME == *"mage"* ]]; then
+  export DATA_HOST=${TARGET_HOSTNAME}
+fi
diff --git a/scripts/custom.funcs.sh b/scripts/custom.funcs.sh
index 060c80b55ae249bd5443bbbb20bcd1a3bedf99e5..ef5b23b3802ba98f4f24d2f5358a485379a74c30 100644
--- a/scripts/custom.funcs.sh
+++ b/scripts/custom.funcs.sh
@@ -99,7 +99,7 @@ services:
       - 8000
     environment:
       - BASE_HREF=${BASE_HREF}
-      - DATA_HOST=${TARGET_HOSTNAME}
+      - DATA_HOST=${DATA_HOST}
       - DATA_PORT=${DATA_PORT}
       - DATA_TYPE=${DATA_TYPE}
       - SITE_URL=${SITE_URL}