diff --git a/GraphNeuralNetworks/test/test_module.jl b/GraphNeuralNetworks/test/test_module.jl
index c3ed96f05..f21e0a298 100644
--- a/GraphNeuralNetworks/test/test_module.jl
+++ b/GraphNeuralNetworks/test/test_module.jl
@@ -31,7 +31,7 @@ using Flux
 using Functors: fmapstructure_with_path
 using Graphs
 using ChainRulesTestUtils, FiniteDifferences
-using Zygote
+using Zygote: Zygote
 using SparseArrays