Fixed final bug with edit button.