ui bug fixing