@Override
  public @Nullable Icon getIcon(final @NotNull PsiElement element, final int _flags) {
    if (element instanceof XmlFile) {
      final VirtualFile vf = ((XmlFile)element).getVirtualFile();
      if (vf != null) {
        final String extension = vf.getExtension();

        if (XSD_FILE_EXTENSION.equals(extension)) {
          return IconManager.getInstance()
            .createLayeredIcon(element, AllIcons.FileTypes.XsdFile, ElementBase.transformFlags(element, _flags));
        }
        if (WSDL_FILE_EXTENSION.equals(extension)) {
          return IconManager.getInstance()
            .createLayeredIcon(element, AllIcons.FileTypes.WsdlFile, ElementBase.transformFlags(element, _flags));
        }
      }
    }
    return null;
  }
@Override
  public @Nullable Icon getIcon(final @NotNull PsiElement element, final int _flags)