Fix UI

Signed-off-by: ayysir <dresadd09691@gmail.com>