Fixes reboot bug