UI Bug fix.