diff --git a/xml/impl/src/com/intellij/lang/html/HtmlFormattingModelBuilder.java b/xml/impl/src/com/intellij/lang/html/HtmlFormattingModelBuilder.java index 7565fe6ad9da..270b69e674b0 100644 --- a/xml/impl/src/com/intellij/lang/html/HtmlFormattingModelBuilder.java +++ b/xml/impl/src/com/intellij/lang/html/HtmlFormattingModelBuilder.java @@ -28,17 +28,19 @@ import com.intellij.psi.codeStyle.CodeStyleSettings; import com.intellij.psi.formatter.FormattingDocumentModelImpl; import com.intellij.psi.formatter.xml.HtmlPolicy; import com.intellij.psi.formatter.xml.XmlBlock; +import com.intellij.psi.impl.source.SourceTreeToPsiMap; import org.jetbrains.annotations.NotNull; public class HtmlFormattingModelBuilder implements FormattingModelBuilder { @Override @NotNull - public FormattingModel createModel(final PsiElement element, final CodeStyleSettings settings) { + public FormattingModel createModel(final PsiElement element, final CodeStyleSettings settings) { final PsiFile psiFile = element.getContainingFile(); final FormattingDocumentModelImpl documentModel = FormattingDocumentModelImpl.createOn(psiFile); return new XmlFormattingModel(psiFile, - new XmlBlock(element.getNode(), - null, null, new HtmlPolicy(settings, documentModel), null, null, false), - documentModel); + new XmlBlock(SourceTreeToPsiMap.psiElementToTree(psiFile), + null, null, new HtmlPolicy(settings, documentModel), null, null, false), + documentModel); } + }