Fix the graph a bit