diff --git a/.github/workflows/unit-tests.yml b/.github/workflows/unit-tests.yml
index 6fec7fcb9a..84f8285f7d 100644
--- a/.github/workflows/unit-tests.yml
+++ b/.github/workflows/unit-tests.yml
@@ -22,7 +22,7 @@ jobs:
         submodules: 'recursive'
     - uses: actions/setup-python@v5
       with:
-        python-version: '3.10'
+        python-version: '3.12'
         cache: 'pip'
     - run: pip install -r requirements-dev.txt
     - run: ./ci/scripts/static_checks.sh
@@ -55,6 +55,12 @@ jobs:
         with:
           images: nemoci.azurecr.io/bionemo
           labels: nemo.library=bionemo
+          tags: |
+            type=schedule
+            type=ref,event=branch
+            type=ref,event=tag
+            type=ref,event=pr
+            type=raw,value=${{ github.run_id }}
 
       - name: Build and push
         uses: docker/build-push-action@v5