Fixing silly UI bug