Minor patch to fix reboot.