diff --git a/.github/workflows/pytest.yml b/.github/workflows/pytest.yml
index 0f93bf1..669751a 100644
--- a/.github/workflows/pytest.yml
+++ b/.github/workflows/pytest.yml
@@ -29,7 +29,7 @@ jobs:
         run: |
           python --version
           python -m pip install --upgrade pip setuptools wheel
-          pip install --upgrade coveralls pytest-cov pytest-flake8 pytest-mypy pytest
+          pip install --upgrade coveralls pytest-cov pytest-flake8 pytest-mypy pytest types-PyYAML
           pip install --upgrade -r requirements.txt
 
       - name: Prepare logfile
@@ -39,7 +39,7 @@ jobs:
 
       - name: Test using pytest
         run: |
-          python3 setup.py pytest
+          python setup.py pytest
 
       - name: Upload pytest test results
         uses: actions/upload-artifact@v2