fix for reboot code