diff --git a/.github/actions/setup-build/action.yml b/.github/actions/setup-build/action.yml
index 8dca54fbde4..9330dc73c74 100644
--- a/.github/actions/setup-build/action.yml
+++ b/.github/actions/setup-build/action.yml
@@ -10,7 +10,7 @@ runs:
       with:
         path: 'aws-kotlin-repo-tools'
         repository: 'awslabs/aws-kotlin-repo-tools'
-        ref: '0.2.3'
+        ref: '0.4.0'
         sparse-checkout: |
           .github
     - name: Checkout smithy-kotlin
diff --git a/.github/workflows/artifact-size-check.yml b/.github/workflows/artifact-size-check.yml
index 9e6d5d4238a..d66de13e8ed 100644
--- a/.github/workflows/artifact-size-check.yml
+++ b/.github/workflows/artifact-size-check.yml
@@ -24,13 +24,8 @@ jobs:
     if: ${{ github.event_name == 'pull_request' }}
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
-      - name: Configure JDK
-        uses: actions/setup-java@v3
-        with:
-          distribution: 'corretto'
-          java-version: 17
-          cache: 'gradle'
+      - name: Setup Build
+        uses: ./aws-sdk-kotlin/.github/actions/setup-build
       - name: Check artifact sizes
         run: | 
           ./gradlew :codegen:sdk:bootstrap -Paws.services=${{ env.BOOTSTRAP_SERVICES }}
@@ -60,13 +55,8 @@ jobs:
     if: ${{ github.event_name == 'release' }}
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
-      - name: Configure JDK
-        uses: actions/setup-java@v3
-        with:
-          distribution: 'corretto'
-          java-version: 17
-          cache: 'gradle'
+      - name: Setup Build
+        uses: ./aws-sdk-kotlin/.github/actions/setup-build
       - name: Check artifact sizes
         run: | 
           ./gradlew :codegen:sdk:bootstrap -Paws.services=${{ env.BOOTSTRAP_SERVICES }}