diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index 39d8a85..ccd3311 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -3,13 +3,17 @@ name: Main build
 on:
   workflow_dispatch:
   workflow_call:
+    inputs:
+      branch:
+        required: false
+        type: string
   push:
     branches: [main]
 
 env:
   REGISTRY: ghcr.io
   NAMESPACE: galasa-dev
-  BRANCH: main
+  BRANCH: ${{ inputs.branch }} || ${{ github.ref_name }}
 
 jobs:
   build-gradle:
@@ -91,6 +95,8 @@ jobs:
         run: |
           docker run --env ARGOCD_AUTH_TOKEN=${{ env.ARGOCD_AUTH_TOKEN }} --rm -v ${{ github.workspace }}:/var/workspace ghcr.io/galasa-dev/argocdcli:main app wait gh-maven-repos --resource apps:Deployment:gradle-gh --health --server argocd.galasa.dev
 
-  call-maven-workflow:
-    needs: build-gradle
-    uses: galasa-dev/maven/.github/workflows/build.yaml@main
\ No newline at end of file
+  # call-maven-workflow:
+  #   needs: build-gradle
+  #   uses: galasa-dev/maven/.github/workflows/build.yaml@main
+  #   with:
+  #     branch: ${{ env.BRANCH }}
\ No newline at end of file