Fix outdated API