Use the simpler await() helper in tests.