diff --git a/GNNLux/src/layers/basic.jl b/GNNLux/src/layers/basic.jl
index 1780d7b5c..0f84bea49 100644
--- a/GNNLux/src/layers/basic.jl
+++ b/GNNLux/src/layers/basic.jl
@@ -35,7 +35,7 @@ julia> m = GNNChain(GCNConv(2=>5),
                     x -> relu.(x), 
                     Dense(5=>4))
 
-julia> x = randn(Float32, 2, 3);
+julia> x = randn(rng, Float32, 2, 3);
 
 julia> g = rand_graph(3, 6)
 GNNGraph: