Fixed reboot functionality.