UI fixed bugs