[FEATURE] Update to use new endpoints.