Update tests to work with latest library.