small fix to UI