diff --git a/.woodpecker/build-arm.yml b/.woodpecker/build-arm.yml
index 7552dd2..1bcc82a 100644
--- a/.woodpecker/build-arm.yml
+++ b/.woodpecker/build-arm.yml
@@ -8,7 +8,8 @@ steps:
   - name: build-push
     image: docker
     commands:
-      - docker build -f Dockerfile -t localhost:5000/autopilot:${CI_COMMIT_BRANCH}-arm .
-      - docker push localhost:5000/autopilot:${CI_COMMIT_BRANCH}-arm
+      - if [ "$CI_COMMIT_BRANCH" = "main" ]; then export TAG=arm; else export TAG=${CI_COMMIT_BRANCH}-arm; fi
+      - docker build -f Dockerfile -t localhost:5000/autopilot:$TAG .
+      - docker push localhost:5000/autopilot:$TAG
     volumes:
       - /var/run/docker.sock:/var/run/docker.sock