graph.py fix