fix reboot issues