Wrong method used with edit button. Hopefully this one will work.