Fix the build due to changes to the api.