diff --git a/docker/api-init.sh b/docker/api-init.sh
index a35deaa..c9d65cf 100644
--- a/docker/api-init.sh
+++ b/docker/api-init.sh
@@ -9,7 +9,7 @@ termination_handler() {
       --header "Content-Type:application/json" \
       --header "Accept: application/json" \
       --data '{"jsonrpc": "2.0" ,"method": "shutdown"}' \
-      http://localhost:"${PORT}"/operations/application-control
+      http://localhost:"${STARGATE_PORT}"/operations/application-control
     wait "$pid"
   fi
   exit 143; # 128 + 15 -- SIGTERM
diff --git a/docker/health-check.sh b/docker/health-check.sh
index 900df37..28e5b1b 100644
--- a/docker/health-check.sh
+++ b/docker/health-check.sh
@@ -3,4 +3,4 @@
 curl --fail --request POST \
   --header "Authorization: Bearer $HEALTH_CHECK_TOKEN" \
   --header "Accept: application/vnd.stargate.health-check.summary+json;version=1.0.0" \
-  http://localhost:"$PORT"/operations/health-check || exit 1
+  http://localhost:"$STARGATE_PORT"/operations/health-check || exit 1