correction bug UI