add restart_done hook