fix little UI