Added missing await keywords