Fix editable button bug