Small fix in graph_test.go