Revert "fix wait"

This reverts commit 4e682cc64ab2c452f5e4891a4d1fe7810d4fd278.