diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 17d3c57..99dfe07 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -20,11 +20,11 @@ jobs:
       matrix:
         include:
           - os: ubuntu-latest
-            python-version: "3.9"
+            python-version: "3.10"
             install-method: mamba
 
           - os: ubuntu-latest
-            python-version: "3.10"
+            python-version: "3.11"
             install-method: mamba
             extra-args: ["codecov"]
 
@@ -44,7 +44,7 @@ jobs:
           PYTHON_VERSION: ${{ matrix.python-version }}
         run: |
           # setup correct python version
-          sed -i -e "s/- python=.*/- python=$PYTHON_VERSION/g" environment.yml
+          sed -i -e "s/- python.*/- python=$PYTHON_VERSION/g" environment.yml
 
       - name: mamba setup
         if: matrix.install-method == 'mamba'
diff --git a/environment.yml b/environment.yml
index e6bbc29..6e2c2d4 100644
--- a/environment.yml
+++ b/environment.yml
@@ -5,7 +5,7 @@ channels:
     - defaults
     - conda-forge
 dependencies:
-  - python < 3.12
+  - python
   - pytorch
   - numpy
   - pip