little change in wait
fix in wait