Fixed a bug in the graph