diff --git a/.github/workflows/image-build.yml b/.github/workflows/image-build.yml
index 7dcf3700..2e57751e 100644
--- a/.github/workflows/image-build.yml
+++ b/.github/workflows/image-build.yml
@@ -19,7 +19,7 @@ permissions:
 jobs:
   docker-image:
     name: Check docker image build
-    runs-on: ubuntu-latest
+    runs-on: ${{ inputs.platform == 'linux/arm64' && 'ubuntu-24.04-arm' || 'ubuntu-latest' }}
     env:
       IMAGE_NAME: stacklok/codegate
       IMAGE_TAG: dev