private void setLeafState(@NotNull LeafState leafState) {
      this.leafState = leafState;
      this.children = leafState == LeafState.ALWAYS ? null : emptyList();
      this.loading = null;
    }
private void setLeafState(@NotNull LeafState leafState)