UI minor change

UI minor change