bug fixing in UI