Fixed the graph viewer