diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml
index 294f58d..2c5dffb 100644
--- a/.github/workflows/test.yaml
+++ b/.github/workflows/test.yaml
@@ -50,7 +50,7 @@ jobs:
       - name: Install test dependencies
         run: |
           python -m pip install --upgrade pip wheel
-          pip install torch>=2.1.0
+          pip install "torch>=2.1.0"
           pip install torch-scatter torch-sparse torch-cluster
       - name: Install dependencies
         run: |
diff --git a/pyproject.toml b/pyproject.toml
index 74f8910..86fda59 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -22,7 +22,7 @@ urls.Source = "https://github.com/theislab/geome"
 urls.Home-page = "https://github.com/theislab/geome"
 dependencies = [
     "anndata",
-    "torch",
+    "torch>=2.1.0",
     "torch_geometric",
     "torch_sparse",
     "torch_cluster",