diff --git a/GNNGraphs/Project.toml b/GNNGraphs/Project.toml
index e0d959e46..5baf95bf0 100644
--- a/GNNGraphs/Project.toml
+++ b/GNNGraphs/Project.toml
@@ -32,7 +32,7 @@ CUDA = "5"
 ChainRulesCore = "1"
 Functors = "0.5"
 Graphs = "1.4"
-KrylovKit = "0.8"
+KrylovKit = "0.8, 0.9"
 LinearAlgebra = "1"
 MLDataDevices = "1.0"
 MLDatasets = "0.7.18"