Fix for UI display.