Added reboot fix