Fixed display error in model