minor bugfix on the graph