Update old code to work with quiz-entity.