[Fix] [Minor] Updated to return response