diff --git a/pyproject.toml b/pyproject.toml
index 10b1eb2..e8fd76b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -31,7 +31,7 @@ dependencies = [
     'torch',
     'torcheval',
     'torchio',
-    'torch_geometric',
+    'torch_geometric' == 2.5.3,
     'tqdm',
     'zarr',
 ]