Fixed Mock code issue for Reboot action