Changes to see if task attempt Id can be used to identify the task type