diff --git a/.drone.yml b/.drone.yml
index 1b35edc..0bb6ae6 100644
--- a/.drone.yml
+++ b/.drone.yml
@@ -64,8 +64,12 @@ steps:
   - name: build-image
     image: docker:24
     environment:
-      API_BASE_URL:
-        from_secret: API_BASE_URL
+      WS_HOST:
+        from_secret: WS_HOST
+      WS_PORT:
+        from_secret: WS_PORT
+      WS_SKEY:
+        from_secret: WS_SKEY
     volumes:
       - name: dockersock
         path: /var/run/docker.sock
@@ -75,9 +79,9 @@ steps:
       - echo "🔨 Building Docker image lila-games/tic-tac-toe-ui:$IMAGE_TAG ..."
       - |
         docker build --network=host \
-        --build-arg VITE_WS_HOST="$API_BASE_URL" \
-        --build-arg VITE_WS_PORT="$API_BASE_URL" \
-        --build-arg VITE_WS_SKEY="$API_BASE_URL" \
+        --build-arg VITE_WS_HOST="$WS_HOST" \
+        --build-arg VITE_WS_PORT="$WS_PORT" \
+        --build-arg VITE_WS_SKEY="$WS_SKEY" \
         -t lila-games/tic-tac-toe-ui:$IMAGE_TAG \
         -t lila-games/tic-tac-toe-ui:latest \
         /drone/src