Fixed compilation error due to TreeMapper and TaskDataSource API change