diff --git a/Project.toml b/Project.toml
index 7155f9d..25af230 100644
--- a/Project.toml
+++ b/Project.toml
@@ -36,8 +36,9 @@ Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 cuDNN = "02a925ec-e4fe-4b08-9a7e-0d78e3d38ccd"
 
-[sources]
-GraphNetCore = {rev = "GNS-small-graph", url = "https://github.com/una-auxme/GraphNetCore.jl"}
+[sources.GraphNetCore]
+rev = "GNS-small-graph"
+url = "https://github.com/una-auxme/GraphNetCore.jl"
 
 [compat]
 Adapt = "4.3.0"
@@ -72,7 +73,7 @@ SciMLSensitivity = "7.76 - 7"
 Statistics = "1"
 Test = "1"
 Zygote = "0.6"
-cuDNN = "1.4 - 1"
+cuDNN = "1.4 - 1, 6"
 julia = "1.10"
 
 [extras]