1 files changed, 9 insertions, 2 deletions
diff --git a/ecs/src/component/storage/graph.rs b/ecs/src/component/storage/graph.rs
index feac16c..e1f48f7 100644
--- a/ecs/src/component/storage/graph.rs
+++ b/ecs/src/component/storage/graph.rs
@@ -381,14 +381,21 @@ impl<'graph> StreamingIterator for ArchetypeAddEdgeDfsIter<'graph>
                 .into_iter(),
         ));
 
-        Some(ArchetypeAddEdgeDfsIterResult::AddEdge(add_edge))
+        Some(ArchetypeAddEdgeDfsIterResult::AddEdge {
+            add_edge_archetype_id: add_edge,
+            add_edge_component_id: component_id,
+        })
     }
 }
 
 #[derive(Debug)]
 pub enum ArchetypeAddEdgeDfsIterResult<'graph, 'iter>
 {
-    AddEdge(ArchetypeId),
+    AddEdge
+    {
+        add_edge_archetype_id: ArchetypeId,
+        add_edge_component_id: Uid,
+    },
     NoEdgesLeftForArchetype,
     NoAddEdge,
     AddEdgeAlreadyVisited,