Trying to call `task_done` the right number of times