@Override
  public void updateCustomComponent(@NotNull JComponent component,
                                    @NotNull Presentation presentation) {
    if (component instanceof ThreeStateCheckBox checkBox) {
      CheckboxAction.updateCheckboxPresentation(checkBox, presentation);

      ThreeStateCheckBox.State property = presentation.getClientProperty(STATE_PROPERTY);
      checkBox.setState(ObjectUtils.chooseNotNull(property, ThreeStateCheckBox.State.NOT_SELECTED));
    }
  }
@Override
  public void updateCustomComponent(@NotNull JComponent component,
                                    @NotNull Presentation presentation)