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);
}
+
}