Fix references to outdated API