fixed UI Bug on displaying after Edit