@Nullable
  private static PyAstLoopStatement getCorrespondingLoopImpl(@NotNull PsiElement element) {
    final PyAstLoopStatement loop = PsiTreeUtil.getParentOfType(element, PyAstLoopStatement.class, true, AstScopeOwner.class);

    if (loop instanceof PyAstStatementWithElse && PsiTreeUtil.isAncestor(((PyAstStatementWithElse)loop).getElsePart(), element, true)) {
      return getCorrespondingLoopImpl(loop);
    }

    return loop;
  }
@Nullable
  private static PyAstLoopStatement getCorrespondingLoopImpl(@NotNull PsiElement element)