diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 18ceb96..259c304 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -22,10 +22,12 @@ jobs:
           if [[ $GITHUB_REF == refs/tags/* ]]; then
             VERSION=${GITHUB_REF#refs/tags/v}
           fi
+          #if this is a branch, use the branch name instead
+          if [[ $GITHUB_REF == refs/heads/* ]]; then
+            VERSION=${GITHUB_REF#refs/heads/}
+          fi
 
           TAGS="${DOCKER_IMAGE}:${VERSION}"
-          #TAGS="${DOCKER_IMAGE}:${VERSION},${DOCKER_IMAGE}:latest,${DOCKER_IMAGE}:${SHORTREF}"
-          # TAGS="${DOCKER_IMAGE}:latest"
 
           echo ::set-output name=tags::${TAGS}
           echo ::set-output name=docker_image::${DOCKER_IMAGE}