fixup to ui