Fixes Minor UI bug