Wait for reboot
Merge pull request #1078 from psav/fix_wait_reboot

Wait for reboot