Add some logic to TaskRegistration