update for new behavior