diff --git a/.github/workflows/budibase_ci.yml b/.github/workflows/budibase_ci.yml
index dfdb1ecc2c..5775d247cb 100644
--- a/.github/workflows/budibase_ci.yml
+++ b/.github/workflows/budibase_ci.yml
@@ -113,7 +113,7 @@ jobs:
       - name: Check submodule
         run: |
           cd packages/pro
-          if ! git branch --contains $(git log -n 1 --pretty=format:%H) | grep -q chore/pro_as_submodule; then
+          if ! git branch --contains $BASE_BRANCH | grep -q chore/pro_as_submodule; then
             echo "Current commit is not part of main branch"
             exit 1
           else