Fix graph_model