diff --git a/.github/actions/setup-environment/action.yml b/.github/actions/setup-environment/action.yml
index 1009a5775ce6f5..64be01b2a557b6 100644
--- a/.github/actions/setup-environment/action.yml
+++ b/.github/actions/setup-environment/action.yml
@@ -1,5 +1,5 @@
 name: Setup environment
-
+description: "Setup environment"
 inputs:
   identity_provider:
     description: "GCP workload identity provider"
@@ -19,8 +19,8 @@ runs:
       name: Authenticate to Google Cloud
       uses: google-github-actions/auth@v2
       with:
-        workload_identity_provider: "${{ inputs.idp }}"
-        service_account: "${{ inputs.sa }}"
+        workload_identity_provider: "${{ inputs.identity_provider }}"
+        service_account: "${{ inputs.service_account }}"
     - name: 'Set up Cloud SDK'
       uses: 'google-github-actions/setup-gcloud@v2'
       with:
@@ -30,3 +30,8 @@ runs:
       shell: bash
       run: |
         gcloud auth configure-docker eu.gcr.io --quiet
+
+        {
+            echo "LEEWAY_SEGMENT_KEY=${{ inputs.leeway_segment_key }}"
+            echo "LEEWAY_WORKSPACE_ROOT=$GITHUB_WORKSPACE"
+        }  >> "$GITHUB_ENV"