UI fixed
slight ui fix