diff --git a/GNNlib/Project.toml b/GNNlib/Project.toml
index f32e7c362..43fb7cee8 100644
--- a/GNNlib/Project.toml
+++ b/GNNlib/Project.toml
@@ -32,7 +32,7 @@ CUDA = "5"
 ChainRulesCore = "1.24"
 DataStructures = "0.18"
 GNNGraphs = "1.4"
-GPUArraysCore = "0.1"
+GPUArraysCore = "0.1,0.2"
 LinearAlgebra = "1"
 MLUtils = "0.4"
 NNlib = "0.9"