Tweak restart script