Fix restart_host script