add wait
fix