update for new Rails