diff --git a/.github/actions/scala_build/action.yml b/.github/actions/scala_build/action.yml
index 950ae2200..ba2c3676c 100644
--- a/.github/actions/scala_build/action.yml
+++ b/.github/actions/scala_build/action.yml
@@ -12,17 +12,24 @@ runs:
           with:
             java-version: '8'
             distribution: 'zulu'
+        - name: Configure python interpreter
+          uses: actions/setup-python@v2
+          with:
+            python-version: ${{ matrix.python }}
+        - name: Add packaged GDAL dependencies
+          shell: bash
+          run : |
+            pip install databricks-mosaic-gdal==${{ matrix.gdal }}
+            sudo tar -xf /home/runner/.local/lib/python${{ matrix.python }}/site-packages/databricks-mosaic-gdal/resources/gdal-${{ matrix.gdal }}-filetree.tar.xz -C /
+            sudo tar -xhf /home/runner/.local/lib/python${{ matrix.python }}/site-packages/databricks-mosaic-gdal/resources/gdal-${{ matrix.gdal }}-symlinks.tar.xz -C /
         - name: Test and build the scala JAR - skip tests is false
           if: inputs.skip_tests == 'false'
           shell: bash
           run: |
-            pip install databricks-mosaic-gdal==3.4.3
-            sudo tar -xf /home/runner/.local/lib/python3.8/site-packages/databricks-mosaic-gdal/resources/gdal-3.4.3-filetree.tar.xz -C /
-            sudo tar -xhf /home/runner/.local/lib/python3.8/site-packages/databricks-mosaic-gdal/resources/gdal-3.4.3-symlinks.tar.xz -C /
             sudo add-apt-repository ppa:ubuntugis/ubuntugis-unstable
             sudo apt clean && sudo apt -o Acquire::Retries=3 update --fix-missing -y
             sudo apt-get -o Acquire::Retries=3 update -y
-            sudo apt-get -o Acquire::Retries=3 install -y gdal-bin=3.4.3+dfsg-1~focal0 libgdal-dev=3.4.3+dfsg-1~focal0 python3-gdal=3.4.3+dfsg-1~focal0
+            sudo apt-get -o Acquire::Retries=3 install -y gdal-bin=${{ matrix.gdal }}+dfsg-1~focal0 libgdal-dev=${{ matrix.gdal }}+dfsg-1~focal0 python3-gdal=${{ matrix.gdal }}+dfsg-1~focal0
             sudo mvn -q clean install
         - name: Build the scala JAR - skip tests is true
           if: inputs.skip_tests == 'true'