Fixing the issue with the graph