update for task api change