Fixes UI bugs.