Unpatch graph_flow.Flow in identity tests