diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 9d3225f..0c9a416 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -11,6 +11,9 @@ jobs:
     steps:
       - name: Checkout
         uses: actions/checkout@v2
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v1
         
       - name: Log in to Docker Hub
         uses: docker/login-action@v1
@@ -22,7 +25,7 @@ jobs:
         uses: docker/login-action@f054a8b539a109f9f41c372932f1ae047eff08c9
         with:
           registry: ghcr.io
-          username: ${{ github.actor }}
+          username: ${{ github.repository_owner }}
           password: ${{ secrets.GITHUB_TOKEN }}
 
       - name: Extract metadata (tags, labels) for Docker
@@ -33,10 +36,10 @@ jobs:
 
       - name: Build and push Docker image
         id: docker_build
-        uses: docker/build-push-action@v1
+        uses: docker/build-push-action@v2
         with:
           context: .
           push: true
           tags: |
-            milesial/unet:latest
-            ghcr.io/milesial/unet:latest
+            milesial/unet:latest 
+            ghcr.io/milesial/pytorch-unet:latest