Add await to prevent throwing exceptions in async mode

Add await to prevent throwing exceptions in async mode #23