Minor update to TaskSet