Update to match changes in the library.