diff --git a/.gitea/workflows/main.yaml b/.gitea/workflows/main.yaml
index f83d953..5150448 100644
--- a/.gitea/workflows/main.yaml
+++ b/.gitea/workflows/main.yaml
@@ -1,5 +1,6 @@
 name: Base DNS Docker Image CI/CD Build
-on: [push]
+on:
+  - push
 
 jobs:
   build-docker-image:
@@ -19,7 +20,7 @@ jobs:
         run: |
           docker push $CONTAINER_NAME:$MAJOR_VERSION.${{ github.run_number }}
       - name: Tag and push Docker image as latest if on master branch
-        if: github.ref == 'refs/heads/master'
+        if: $GITHUB_REF == 'refs/heads/main'
         run: |
           docker tag $CONTAINER_NAME:$MAJOR_VERSION.${{ github.run_number }} $CONTAINER_NAME:latest
           docker push $CONTAINER_NAME:latest