Fix up forget/save citation buttons.