Fix previously mentioned bug(assign_vertex_id)