Add await to response