diff --git a/GNNGraphs/src/transform.jl b/GNNGraphs/src/transform.jl
index d7177a0e8..8e8c98d13 100644
--- a/GNNGraphs/src/transform.jl
+++ b/GNNGraphs/src/transform.jl
@@ -214,14 +214,7 @@ end
 function remove_edges(g::GNNGraph{<:COO_T}, p = 0.5)
     num_edges = g.num_edges
     edges_to_remove = filter(_ -> rand() < p, 1:num_edges)        
-    g = remove_edges(g, edges_to_remove)
-    s, t = edge_index(g)
-    w = get_edge_weight(g)
-    edata = g.edata
-    return GNNGraph((s, t, w),
-             g.num_nodes, length(s), g.num_graphs,
-             g.graph_indicator,
-             g.ndata, edata, g.gdata)
+    return remove_edges(g, edges_to_remove)
 end
 
 """