update for new API changes