Move this inside the await block