public @Nullable TextAttributesKey getAttributesKey(PsiElement element) {
    return getState(element).getEditorAttributesKey();
  }
public @Nullable TextAttributesKey getAttributesKey(PsiElement element)