diff --git a/.github/workflows/docker-image.yml b/.github/workflows/docker-image.yml
index 1ae6a97..ec21352 100644
--- a/.github/workflows/docker-image.yml
+++ b/.github/workflows/docker-image.yml
@@ -39,7 +39,7 @@ jobs:
           type=raw,value=latest,enable=${{ github.ref == format('refs/heads/{0}', 'main') }}
           type=ref,event=branch
           type=semver,pattern={{version}}
-          type=sha,prefix={{branch}}-
+          type=sha,prefix=main-,enable=${{ github.ref == format('refs/heads/{0}', 'main') }}
 
     - name: Set up QEMU
       uses: docker/setup-qemu-action@v2
diff --git a/README.md b/README.md
index aa8f6f6..4d20aaa 100644
--- a/README.md
+++ b/README.md
@@ -1,8 +1,8 @@
 # Custom SSH Server Docker Setup
 
-This repository contains a Docker setup for creating a custom SSH server based on Alpine Linux. The setup includes the ability to use static SSH host keys, configure SSH key and password authentication, and set up user-specific SSH key pairs.
+[![Build](https://github.com/haravich/custom-ssh-server/actions/workflows/docker-image.yml/badge.svg)](https://github.com/haravich/custom-ssh-server/actions/workflows/docker-image.yml)![Docker pulls](https://img.shields.io/docker/pulls/haravich/custom-ssh-server)
 
-[![Build](https://github.com/haravich/custom-ssh-server/actions/workflows/docker-image.yml/badge.svg)](https://github.com/haravich/custom-ssh-server/actions/workflows/docker-image.yml)
+This repository contains a Docker setup for creating a custom SSH server based on Alpine Linux. The setup includes the ability to use static SSH host keys, configure SSH key and password authentication, and set up user-specific SSH key pairs.
 
 ## Table of Contents