Fixed UI switching via context menu

TBR: @dinhviethoa