diff --git a/.github/workflows/ingestion-ci.yml b/.github/workflows/ingestion-ci.yml
index 03df48d9..90e301ad 100644
--- a/.github/workflows/ingestion-ci.yml
+++ b/.github/workflows/ingestion-ci.yml
@@ -18,7 +18,7 @@ jobs:
       fail-fast: false
       matrix:
         os: ["ubuntu-latest"] # , "macos-latest", "windows-latest"
-        python-version: ["3.9"]
+        python-version: ["3.11"]
 
     steps:
       - uses: actions/checkout@v4
diff --git a/.github/workflows/spatial-ci.yml b/.github/workflows/spatial-ci.yml
index 25018a41..e123f9dc 100644
--- a/.github/workflows/spatial-ci.yml
+++ b/.github/workflows/spatial-ci.yml
@@ -18,7 +18,7 @@ jobs:
       fail-fast: false
       matrix:
         os: ["ubuntu-latest"] # , "macos-latest", "windows-latest"
-        python-version: ["3.9"]
+        python-version: ["3.11"]
 
     steps:
       - uses: actions/checkout@v4
diff --git a/pyproject.toml b/pyproject.toml
index 7737305b..5114cce7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -36,7 +36,8 @@ dependencies = [
     "drmaa", 
     "gevent",
     "harmonypy",
-    #removed jax to figure if it's needed by cell2location
+    "jax<=0.4.23", # temp for scvi https://github.com/theislab/scvelo/issues/1186
+    "jaxlib<=0.4.23",
     "leidenalg", 
     "louvain",
     "matplotlib<=3.7.3",
@@ -72,9 +73,7 @@ notebook = [
 spatial = [
     "squidpy",
     "cell2location",
-    "tangram-sc",
-    "jax<=0.4.23",
-    "jaxlib<=0.4.23",
+    "tangram-sc"
 ]
 pypi_upload = [
     "twine",