bug fix, ui refine