Recent ui-bugs fixes