Add driver restart completed handler