diff --git a/.github/workflows/docker-build.yaml b/.github/workflows/docker-build.yaml
index 01b2733..6cb6e30 100644
--- a/.github/workflows/docker-build.yaml
+++ b/.github/workflows/docker-build.yaml
@@ -16,7 +16,7 @@ env:
   REACT_APP_ADMIN_EMAIL: ${{ vars.REACT_APP_ADMIN_EMAIL}}
   REACT_APP_SITE_NAME: ${{ vars.REACT_APP_SITE_NAME}}
   STAGE: ${{ vars.STAGE}}
-  
+
 jobs:
   push_to_registry:
     strategy:
@@ -46,4 +46,5 @@ jobs:
           context: ${{ matrix.service }}
           platforms: linux/arm64,linux/amd64
           push: true
-          tags: ${{ vars.DOCKER_NAMESPACE }}/trunk-server-${{ matrix.service }}:latest
\ No newline at end of file
+          tags: ${{ vars.DOCKER_NAMESPACE }}/trunk-server-${{ matrix.service }}:latest
+          build-args: STAGE=${{vars.STAGE}}
\ No newline at end of file