Fix update task test