@Override
  protected @NotNull PsiElement getTopmostExpression(@NotNull PsiElement element) {
    PsiElement parent = element.getParent();
    if (parent instanceof PsiExpressionStatement) {
      return parent;
    }

    return element;
  }
@Override
  protected @NotNull PsiElement getTopmostExpression(@NotNull PsiElement element)