bug fix
fix ui