Make the total restart work.
Fix the fix :(