Avoiding flaky await semantics in tests