[UI] Fixed method duplication