model display fix after refactor