diff --git a/.github/workflows/deploy-doc.yml b/.github/workflows/deploy-doc.yml
index 77e5f0e..6b7771a 100644
--- a/.github/workflows/deploy-doc.yml
+++ b/.github/workflows/deploy-doc.yml
@@ -3,7 +3,7 @@ on:
   workflow_dispatch: {}
   push:
     branches:
-      - development
+      - main
 jobs:
   deploy:
     runs-on: ubuntu-latest
diff --git a/.github/workflows/prod-build.yml b/.github/workflows/prod-build.yml
index 4763225..8a862fa 100644
--- a/.github/workflows/prod-build.yml
+++ b/.github/workflows/prod-build.yml
@@ -28,6 +28,7 @@ jobs:
       - name: Check out the repo
         uses: actions/checkout@v3
 
+      #Fetch Latest release
       - name: Fetch latest release
         id: fetch-latest-release
         uses: InsonusK/get-latest-release@v1.0.1
@@ -39,18 +40,22 @@ jobs:
         run: |
           echo "tag_name: ${{ steps.fetch-latest-release.outputs.tag_name }}"
 
+      #Login to Docker Hub
       - name: Log in to Docker Hub
         uses: docker/login-action@f4ef78c080cd8ba55a85445d5b36e214a81df20a
         with:
           username: ${{ env.DOCKER_HUB_USERNAME  }}
           password: ${{ env.DOCKER_HUB_PASSWORD  }}
-      
+
+      #Build and push Docker image      
       - name: Build and push Docker image
         uses: docker/build-push-action@v4
         with:
           context: .
           file: ./Dockerfile
           push: true
+          build-args: |
+            RELEASE_VERSION=${{ steps.fetch-latest-release.outputs.tag_name }}
           tags: ${{ env.REPOSITORY_NAMESPACE }}/${{ env.REPOSITORY_NAME }}:${{ steps.fetch-latest-release.outputs.tag_name }}
           username: ${{ env.DOCKER_HUB_USERNAME  }}
           password: ${{ env.DOCKER_HUB_PASSWORD  }}
\ No newline at end of file