diff --git a/.github/workflows/build-vscode-extension.yml b/.github/workflows/build-vscode-extension.yml
index 5479642a1..05cbc7513 100644
--- a/.github/workflows/build-vscode-extension.yml
+++ b/.github/workflows/build-vscode-extension.yml
@@ -23,12 +23,14 @@ jobs:
     steps:
       - name: Checkout vscode-extensions code and workflow scripts
         uses: actions/checkout@v4
+      - name: Set up JDK 17
+        uses: actions/setup-java@v4
+        with:
+          java-version: '17'
+          distribution: 'temurin'
       - name: Build .VSIX file
         id: build-vsix
         run: |
-          java -version
-          echo "::set-env name=JAVA_HOME::$JAVA_HOME_17"
-          java -version
           ${{ github.workspace }}/.github/scripts/build-vscode-extension.sh ${{ inputs.extension-name }} ${{ inputs.dist }}
           ls ./vsix