fix error when edit