Fixes
ui tweak