Fix reboot bug.