Fix misc UI bugs