fixed the graph wrapper class name