moved wait msg out of restore func