Fixed issues in update reboot process