Follow library API change