Use wait module when restarting machine