Fix edit button handling.