But I don't think it belongs in build_graph either...