@Nullable
    @Override
    public PsiElement getContainer(@Nullable PsiElement context) {
      return context == null || !context.isValid() ? null : context.getContainingFile();
    }
@Nullable
    @Override
    public PsiElement getContainer(@Nullable PsiElement context)