diff --git a/GNNLux/test/layers/pool.jl b/GNNLux/test/layers/pool.jl
index 69dee0926..48f9ee65c 100644
--- a/GNNLux/test/layers/pool.jl
+++ b/GNNLux/test/layers/pool.jl
@@ -8,13 +8,13 @@
         x = randn(rng, Float32, in_dims, 10)
 
         @testset "GlobalPool" begin
-            l = GNNLux.GlobalPool(mean)
+            l = GlobalPool(mean)
             test_lux_layer(rng, l, g, x, sizey=(in_dims,1))
         end
         @testset "GlobalAttentionPool" begin
             fgate = Dense(in_dims, 1)
             ffeat = Dense(in_dims, in_dims)
-            l = GNNLux.GlobalAttentionPool(fgate, ffeat)
+            l = GlobalAttentionPool(fgate, ffeat)
             test_lux_layer(rng, l, g, x, sizey=(in_dims,1), container=true)
         end
     end