update for API change