bug fix on ui