diff --git a/GNNLux/src/layers/conv.jl b/GNNLux/src/layers/conv.jl
index 63c4f90b4..3b70747c8 100644
--- a/GNNLux/src/layers/conv.jl
+++ b/GNNLux/src/layers/conv.jl
@@ -1,8 +1,8 @@
 _getbias(ps) = hasproperty(ps, :bias) ? getproperty(ps, :bias) : false
 _getstate(st, name) = hasproperty(st, name) ? getproperty(st, name) : NamedTuple()
-_getstate(s::StatefulLuxLayer{true}) = s.st
+_getstate(s::StatefulLuxLayer{Val{true}}) = s.st
 _getstate(s::StatefulLuxLayer{Static.True}) = s.st
-_getstate(s::StatefulLuxLayer{false}) = s.st_any
+_getstate(s::StatefulLuxLayer{Val{false}}) = s.st_any
 _getstate(s::StatefulLuxLayer{Static.False}) = s.st_any
 
 @doc raw"""
diff --git a/GNNLux/test/Project.toml b/GNNLux/test/Project.toml
index d651808bf..75b82f42d 100644
--- a/GNNLux/test/Project.toml
+++ b/GNNLux/test/Project.toml
@@ -20,3 +20,6 @@ Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 TestItemRunner = "f8b46487-2199-4994-9208-9a1283c18c0a"
 TestItems = "1c621080-faea-4a02-84b6-bbd5e436b8fe"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
+
+[compat]
+LuxTestUtils = "1"