Update graph_test.go