Update library

This behavior won't work now.