Update TaskSetManagerSuite.scala