Updated reboot check