UI Bug fixes.