diff --git a/.github/workflows/integration.yml b/.github/workflows/integration.yml
index f5fa9ff0f..e63010786 100644
--- a/.github/workflows/integration.yml
+++ b/.github/workflows/integration.yml
@@ -168,7 +168,7 @@ jobs:
             ${{ runner.os }}-gradle-
       - name: Install DJL-Bench
         working-directory: benchmark
-        run: ./gradlew installOnLinux
+        run: ./gradlew installOnLinux  -Pstaging
       - name: Build container name
         run: ./serving/docker/scripts/docker_name_builder.sh pytorch-gpu ${{ github.event.inputs.djl-version }}
       - name: Download models and dockers
@@ -226,7 +226,7 @@ jobs:
             ${{ runner.os }}-gradle-
       - name: Install DJL-Bench
         working-directory: benchmark
-        run: ./gradlew installOnLinux
+        run: ./gradlew installOnLinux -Pstaging
       - name: Build container name
         run: ./serving/docker/scripts/docker_name_builder.sh aarch64 ${{ github.event.inputs.djl-version }}
       - name: Download models and dockers