Updated migration logic for TaskTracker API change