fixed Edit UI bug