fixed ui bug