Make reboot on failure conditional