Fix bug that broke model displays