Fixed a minor bug in the ui