diff --git a/.github/actions/setup-docker-environment/action.yaml b/.github/actions/setup-docker-environment/action.yaml
index 6289ed0a..6a181f05 100644
--- a/.github/actions/setup-docker-environment/action.yaml
+++ b/.github/actions/setup-docker-environment/action.yaml
@@ -15,7 +15,10 @@ runs:
       id: vars
       run: |
         echo ::set-output name=sha_short::${GITHUB_SHA::7}
-        TAG=${GITHUB_REF/refs\/tags\//}
+        TAG=${GITHUB_REF##*/}
+        if [ "main" == "${TAG} ]; then
+          TAG=canary
+        fi
         echo ::set-output name=tag::${TAG:-canary}
       shell: bash