diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 28c0848c..5ca70ef1 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -9,15 +9,14 @@ on:
 jobs:
   PyTest:
     runs-on: ubuntu-latest
-    # container: pytorch/pytorch:1.8.1-cuda11.1-cudnn8-runtime
     steps:
       - uses: actions/checkout@v2
       - uses: actions/setup-python@v2
         with:
-          python-version: "3.8"
+          python-version: "3.10"
       - name: Install Dependencies
         run: |
           curl -sSL https://install.python-poetry.org | python3 -
           poetry install
       - name: pytest
-        run: poetry run pytest tests
\ No newline at end of file
+        run: poetry run pytest tests