Minor UI fix for model screen.