diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index e0b451ea6da3eb39ca9404202d128d37d2a2943e..1f4325c40a55154544d7238270051d31737be25b 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -105,6 +105,10 @@ variables:
     DATA_PORT: '2060'
     DATA_TYPE: 'edge'
 
+.mage01:
+  tags:
+    - mage01
+
 .staging:
   only:
     - master@ghsc/geomag/geomag-algorithms
@@ -150,5 +154,6 @@ Deploy Mage01:
     - .deploy
     - .staging
     - .mage
+    - .mage01
   variables:
     APP_NAME: geomag-algorithms
diff --git a/scripts/custom.funcs.sh b/scripts/custom.funcs.sh
index 2a9e7f137c9ec32c8aff18448c4e15e0837af892..9082bc665a1d61e99f9836d063e306699a2e92fb 100644
--- a/scripts/custom.funcs.sh
+++ b/scripts/custom.funcs.sh
@@ -34,7 +34,7 @@ services:
       - 8000
     environment:
       - BASE_HREF=${BASE_HREF}
-      - DATA_HOST=${DATA_HOST}
+      - DATA_HOST=${TARGET_HOSTNAME}
       - DATA_PORT=${DATA_PORT}
       - DATA_TYPE=${DATA_TYPE}
       - SITE_URL=${SITE_URL}