diff --git a/GraphNeuralNetworks/src/GraphNeuralNetworks.jl b/GraphNeuralNetworks/src/GraphNeuralNetworks.jl
index 13030003c..49e1317b1 100644
--- a/GraphNeuralNetworks/src/GraphNeuralNetworks.jl
+++ b/GraphNeuralNetworks/src/GraphNeuralNetworks.jl
@@ -10,6 +10,7 @@ using NNlib: scatter, gather
 using ChainRulesCore
 using Reexport
 using MLUtils: zeros_like
+using Graphs
 
 using GNNGraphs: COO_T, ADJMAT_T, SPARSE_T,
                   check_num_nodes, check_num_edges,
diff --git a/GraphNeuralNetworks/src/samplers.jl b/GraphNeuralNetworks/src/samplers.jl
index 4157b52fc..647edd688 100644
--- a/GraphNeuralNetworks/src/samplers.jl
+++ b/GraphNeuralNetworks/src/samplers.jl
@@ -1,6 +1,3 @@
-using GraphNeuralNetworks
-using Graphs
-
 """
     struct NeighborLoader