Fixes the wait() test helper which was broken since the introduction of the async router