Fixes UI bugs