we were using the wrong source graph in get_collapsed_graph