Fix async-return diagnostics with await

For async_return_with_no_await, add await and preserve the async method.