Update for dependency API change