Fix addEditButton