fix reboot issue