UI methods fix