Forgot updating the call to the graph generator.