diff --git a/pyproject.toml b/pyproject.toml
index d48eb646c..fe530168e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -75,7 +75,7 @@ mpi = ["mpi4py>=4.0.2"]
 
 notebooks = ["py3Dmol>=2.5.2", "marimo>=0.17.6", "jupyter"]
 
-deeprank-gnn-esm = ["deeprank-gnn-esm==1.1.0"]
+deeprank-gnn-esm = ["deeprank-gnn-esm==2.0.0"]
 
 [project.urls]
 Homepage = "https://github.com/haddocking/haddock3"