fixed the tests for the graph