bug fix at edit controller