Fix more ui stuff.