diff --git a/.github/actions/setup-python/action.yaml b/.github/actions/setup-python/action.yaml
index 59c65d9..8749c5f 100644
--- a/.github/actions/setup-python/action.yaml
+++ b/.github/actions/setup-python/action.yaml
@@ -1,9 +1,15 @@
 name: Setup Python
 description: "Installs Python"
 
+inputs:
+  python_version:
+    description: "The version of Python to install"
+    required: true
+    default: "3.11"
+
 runs:
   using: "composite"
   steps:
-    - uses: actions/setup-python@v2
+    - uses: actions/setup-python@v4
       with:
-        python-version: "3.10"
+        python-version: ${{ inputs.python_version }}
diff --git a/.github/workflows/unit-tests.yaml b/.github/workflows/unit-tests.yaml
index 35d7000..a5b1d92 100644
--- a/.github/workflows/unit-tests.yaml
+++ b/.github/workflows/unit-tests.yaml
@@ -10,7 +10,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - name: checkout
-        uses: actions/checkout@v2
+        uses: actions/checkout@v4
       - name: setup python
         uses: ./.github/actions/setup-python
       - run: make venv-install
@@ -21,7 +21,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - name: checkout
-        uses: actions/checkout@v2
+        uses: actions/checkout@v4
       - name: setup python
         uses: ./.github/actions/setup-python
       - run: make venv-install
@@ -37,10 +37,11 @@ jobs:
     runs-on: ubuntu-latest
     if: github.ref == 'refs/heads/main'
     steps:
-      - uses: actions/checkout@v3
-      - uses: actions/setup-python@v4
+      - uses: actions/checkout@v4
+      - name: setup python
+        uses: ./.github/actions/setup-python
         with:
-          python-version: "3.10"
+          python_version: "3.10"
       - name: Download Coverage Report
         uses: actions/download-artifact@v3
         with: