Update for API
Update to match API