diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml
index 3e3708b045..51cb58fa8b 100644
--- a/.github/workflows/docker-build.yml
+++ b/.github/workflows/docker-build.yml
@@ -4,6 +4,7 @@ on:
   push:
     branches:
       - master
+      - improve-docker-build
     paths:
       - '.github/workflows/docker-build.yml'
       - 'docker/**'
@@ -64,6 +65,9 @@ jobs:
         mkdir -p $DOCKER_CONFIG/cli-plugins
         curl -SL https://github.com/docker/buildx/releases/download/v0.11.2/buildx-v0.11.2.darwin-arm64 -o $DOCKER_CONFIG/cli-plugins/docker-buildx
         chmod +x $DOCKER_CONFIG/cli-plugins/docker-buildx
+    - name: Setup tmate session
+      if: runner.os == 'macos'
+      uses: mxschmitt/action-tmate@v3
     - env:
         SPARK_VERSION: ${{ matrix.spark }}
         SEDONA_VERSION: ${{ matrix.sedona }}