diff --git swh/dataset/graph.py swh/dataset/graph.py
index 52165fe..51e9bb9 100644
--- swh/dataset/graph.py
+++ swh/dataset/graph.py
@@ -46,7 +46,7 @@ def process_messages(messages, config, node_writer, edge_writer, node_set):
 
     messages = {k: fix_objects(k, v) for k, v in messages.items()}
 
-    for visit in messages.get("origin_visit", []):
+    for visit in messages.get("origin_visit_status", []):
         origin_id = origin_identifier({"url": visit["origin"]})
         visit_id = visit["visit"]
         if not node_set.add("{}:{}".format(origin_id, visit_id).encode()):