small modify in DAGScheduler