diff --git a/.github/workflows/budibase_ci.yml b/.github/workflows/budibase_ci.yml
index 68cf48be5c..01ecbecac8 100644
--- a/.github/workflows/budibase_ci.yml
+++ b/.github/workflows/budibase_ci.yml
@@ -96,9 +96,10 @@ jobs:
       - name: Check submodule
         run: |
           cd packages/pro
-          if ! git branch --contains $GITHUB_BASE_REF | grep -q chore/pro_as_submodule; then
+          if ! git branch --contains $github.head_ref | grep -q chore/pro_as_submodule; then
             echo "Current commit is not part of main branch"
             exit 1
           else
             echo "All good, the submodule has been merged!"
           fi
+-
\ No newline at end of file