diff --git a/.github/workflows/unit-tests.yml b/.github/workflows/unit-tests.yml
index f0c63f83d..b133511d1 100644
--- a/.github/workflows/unit-tests.yml
+++ b/.github/workflows/unit-tests.yml
@@ -35,6 +35,8 @@ jobs:
 
       - name: Install dependencies
         uses: ./.github/actions/setup-runner
+        with:
+          python-version: ${{ matrix.python }}
 
       - name: Run unit tests
         run: |
diff --git a/scripts/unit-tests.sh b/scripts/unit-tests.sh
index 9e271e350..ff42d3039 100755
--- a/scripts/unit-tests.sh
+++ b/scripts/unit-tests.sh
@@ -5,6 +5,9 @@
 #
 # This source code is licensed under the terms described in the LICENSE file in
 # the root directory of this source tree.
+
+PYTHON_VERSION=${PYTHON_VERSION:-3.12}
+
 set -e
 
 # Always run this at the end, even if something fails
@@ -14,8 +17,6 @@ cleanup() {
 }
 trap cleanup EXIT
 
-PYTHON_VERSION=${PYTHON_VERSION:-3.12}
-
 command -v uv >/dev/null 2>&1 || { echo >&2 "uv is required but it's not installed. Exiting."; exit 1; }
 
 uv python find "$PYTHON_VERSION"
diff --git a/tests/unit/providers/utils/test_model_registry.py b/tests/unit/providers/utils/test_model_registry.py
index f74abbfba..db1630000 100644
--- a/tests/unit/providers/utils/test_model_registry.py
+++ b/tests/unit/providers/utils/test_model_registry.py
@@ -179,6 +179,7 @@ async def test_register_model_existing_different(
 async def test_register_model_during_init(helper: ModelRegistryHelper, known_model: Model) -> None:
     assert helper.get_provider_model_id(known_model.provider_resource_id) == known_model.provider_model_id
 
+
 # TODO: unregister_model functionality was removed/disabled by https://github.com/meta-llama/llama-stack/pull/2916
 # async def test_unregister_model_during_init(helper: ModelRegistryHelper, known_model: Model) -> None:
 #     assert helper.get_provider_model_id(known_model.provider_resource_id) == known_model.provider_model_id