fixing more ui elements