Update library to fix broken tests