Fixed tiny bug in the ui