private static @Nullable PsiElement unwrapStatementListContainer(@Nullable PsiElement element) {
      return element instanceof PyStatementListContainer ? ((PyStatementListContainer)element).getStatementList() : element;
    }
private static @Nullable PsiElement unwrapStatementListContainer(@Nullable PsiElement element)