I think it would be better to make stageId a property of a task set manager and remove it from the schedulable interface