diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 8cf187b7..8036417a 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -26,6 +26,9 @@ jobs:
           - os: windows-latest
             torch-version: 2.0.0
             python-version: "3.10"
+          - os: ubuntu-latest
+            torch-version: 2.1.1
+            python-version: "3.11"
           #- os: macos-latest
           #  torch-version: 2.0.0
           #  python-version: "3.10"
diff --git a/conda/cebra.yml b/conda/cebra.yml
index 6b65e246..7bc9eb7c 100644
--- a/conda/cebra.yml
+++ b/conda/cebra.yml
@@ -17,10 +17,9 @@ channels:
     - pytorch
     - defaults
 dependencies:
-    - python<3.10
+    - python
     - pip
     - pytorch
     - jupyter
-    - nb_conda
     - pip:
         - cebra
diff --git a/conda/cebra_paper_m1.yml b/conda/cebra_paper_m1.yml
index 223126f4..16be7c2c 100644
--- a/conda/cebra_paper_m1.yml
+++ b/conda/cebra_paper_m1.yml
@@ -21,7 +21,7 @@
 # source ~/miniconda/bin/activate
 # conda init zsh
 
-name: piVAE_m1
+name: cebra_paper_m1
 channels:
     - apple
     - pytorch-lts