Modify the reboot method.