update reboot logic for mlnx