trying to fix the graph