Migrate put_graph to new signature

Not much to say here, really.