Fixes UI bug