Fix reboot bug