fix ui
Merge pull request #1372 from bechi/feature/fix-ui

fix ui