small fix
more UI related fixes