diff --git a/.github/workflows/attach-artifact-release.yml b/.github/workflows/attach-artifact-release.yml
index 94015df3..63093c5e 100644
--- a/.github/workflows/attach-artifact-release.yml
+++ b/.github/workflows/attach-artifact-release.yml
@@ -7,6 +7,6 @@ on:
 
 jobs:
   attach-artifact-to-release:
-    uses: liquibase/build-logic/.github/workflows/extension-attach-artifact-release.yml@v0.7.8
+    uses: liquibase/build-logic/.github/workflows/extension-attach-artifact-release.yml@main
     secrets: inherit
 
diff --git a/.github/workflows/build-nightly.yml b/.github/workflows/build-nightly.yml
index c9f9bed9..17ed05c5 100644
--- a/.github/workflows/build-nightly.yml
+++ b/.github/workflows/build-nightly.yml
@@ -7,7 +7,7 @@ on:
 
 jobs:
     nightly-build:
-      uses: liquibase/build-logic/.github/workflows/os-extension-test.yml@v0.7.8
+      uses: liquibase/build-logic/.github/workflows/os-extension-test.yml@main
       with:
         nightly: true
       secrets: inherit
diff --git a/.github/workflows/create-release.yml b/.github/workflows/create-release.yml
index 5280eeb4..d1db6d78 100644
--- a/.github/workflows/create-release.yml
+++ b/.github/workflows/create-release.yml
@@ -7,6 +7,6 @@ on:
 
 jobs:
   create-release:
-    uses: liquibase/build-logic/.github/workflows/create-release.yml@v0.7.8
+    uses: liquibase/build-logic/.github/workflows/create-release.yml@main
     secrets: inherit
 
diff --git a/.github/workflows/documentation.yml b/.github/workflows/documentation.yml
index 045e8611..c54af56b 100644
--- a/.github/workflows/documentation.yml
+++ b/.github/workflows/documentation.yml
@@ -17,6 +17,8 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
+        with:
+          ref: ${{ github.event.pull_request.head.sha || github.ref }}
 
       - name: Build
         run: |
diff --git a/.github/workflows/release-published.yml b/.github/workflows/release-published.yml
index 68e512e0..4fd4d917 100644
--- a/.github/workflows/release-published.yml
+++ b/.github/workflows/release-published.yml
@@ -7,7 +7,7 @@ on:
 
 jobs:
   release:
-    uses: liquibase/build-logic/.github/workflows/extension-release-published.yml@v0.7.8
+    uses: liquibase/build-logic/.github/workflows/extension-release-published.yml@main
     secrets: inherit
     with:
-      nameSpace: neo4j
\ No newline at end of file
+      nameSpace: neo4j
diff --git a/.github/workflows/test-5x-driver.yml b/.github/workflows/test-5x-driver.yml
index 2f182b2b..2eebf843 100644
--- a/.github/workflows/test-5x-driver.yml
+++ b/.github/workflows/test-5x-driver.yml
@@ -22,7 +22,7 @@ jobs:
         
   build-test:
     needs: authorize
-    uses: liquibase/build-logic/.github/workflows/os-extension-test.yml@v0.7.8
+    uses: liquibase/build-logic/.github/workflows/os-extension-test.yml@main
     secrets: inherit
 
   integration-test:
@@ -63,6 +63,8 @@ jobs:
 
     steps:
       - uses: actions/checkout@v4
+        with:
+          ref: ${{ github.event.pull_request.head.sha || github.ref }}
 
       - name: Set up JDK ${{ matrix.java }}
         uses: actions/setup-java@v4
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 8def8104..8d746fd1 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -26,7 +26,7 @@ jobs:
       
   build-test:
     needs: authorize
-    uses: liquibase/build-logic/.github/workflows/os-extension-test.yml@v0.7.8
+    uses: liquibase/build-logic/.github/workflows/os-extension-test.yml@main
     secrets: inherit
 
   integration-test:
@@ -72,6 +72,8 @@ jobs:
 
     steps:
       - uses: actions/checkout@v4
+        with:
+          ref: ${{ github.event.pull_request.head.sha || github.ref }}
 
       - name: Set up JDK ${{ matrix.java }}
         uses: actions/setup-java@v4