Fix bug with edit function