Update to match the current API
GA update