diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index ab22cbb2..c11b97a8 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -130,7 +130,7 @@ jobs:
       uses: actions/cache@v3
       with:
         path: container-test-template-cache
-        key: ${{ runner.os }}-test-template-cache-${{ hashFiles('containers/test-template/**') }}
+        key: ${{ runner.os }}-test-template-cache-${{ hashFiles('containers/test-template/**') }}-${{ matrix.python-version }}
 
     - name: Inject container-test-template-cache into docker
       uses: reproducible-containers/buildkit-cache-dance@v2.1.2
@@ -144,8 +144,10 @@ jobs:
         cache-to: type=gha,mode=max
         push: false
         load: true
-        tags: snakebids/test-template:dev
+        tags: snakebids/test-template:${{ matrix.python-version }}
         platforms: linux/amd64
+        build-args: |
+          PYTHON_VERSION=${{ matrix.python-version }}
     #----------------------------------------------
     # install your root project, if required
     #----------------------------------------------