@Nullable
  private static PsiElement getControlStatementHeader(@NotNull ASTNode node) {
    final PyAstStatementPart statementPart = PsiTreeUtil.getParentOfType(node.getPsi(), PyAstStatementPart.class, false, PyAstStatementList.class);
    if (statementPart != null) {
      return statementPart;
    }
    final PyAstWithItem withItem = PsiTreeUtil.getParentOfType(node.getPsi(), PyAstWithItem.class);
    if (withItem != null) {
      return withItem.getParent();
    }
    return null;
  }
@Nullable
  private static PsiElement getControlStatementHeader(@NotNull ASTNode node)