Update TaskSchedulerImpl.scala