diff --git a/.github/workflows/code_checks.yml b/.github/workflows/code_checks.yml
index 1f2f81aa..e14373d2 100755
--- a/.github/workflows/code_checks.yml
+++ b/.github/workflows/code_checks.yml
@@ -46,9 +46,9 @@ jobs:
         uses: actions/checkout@v4
 
       - name: Set up Python
-        uses: actions/setup-python@v4
+        uses: actions/setup-python@v5
         with:
-          python-version: '3.x'
+          python-version: '3.11'
 
       - name: Install linting tools
         run: |
@@ -94,15 +94,16 @@ jobs:
       - name: Set up Python
         uses: actions/setup-python@v4
         with:
-          python-version: '3.x'
+          python-version: '3.11'
 
       - name: Install dependencies
         run: |
           pip install -r requirements.txt
-          pip install pytest
+          pip install pytest pyyaml
 
       - name: Run unit tests
         run: |
           echo "🧪 Running unit tests..."
+          export PYTHONPATH=$PYTHONPATH:./server
           pytest -m "not (docker or compose or feature_complete)"