private static PsiStatement findStatement(@NotNull ControlFlow flow, int offset) {
    PsiElement element = flow.getElement(offset);
    return PsiTreeUtil.getParentOfType(element, PsiStatement.class, false);
  }
private static PsiStatement findStatement(@NotNull ControlFlow flow, int offset)