fix reboot function