Fix bug with reboot after update