update for image api change