small fix to ui