Update TaskSetManager.scala