Update driver to use the new TaskNode API.