Minor fix on tool UI