diff --git a/python/python-rest/src/com/jetbrains/rest/RestFileViewProvider.java b/python/python-rest/src/com/jetbrains/rest/RestFileViewProvider.java
index c9d153fbc973..816e46158100 100644
--- a/python/python-rest/src/com/jetbrains/rest/RestFileViewProvider.java
+++ b/python/python-rest/src/com/jetbrains/rest/RestFileViewProvider.java
@@ -10,6 +10,7 @@ import com.intellij.psi.PsiFile;
 import com.intellij.psi.PsiManager;
 import com.intellij.psi.impl.source.PsiFileImpl;
 import com.intellij.psi.templateLanguages.TemplateLanguageFileViewProvider;
+import com.intellij.psi.tree.IElementType;
 import com.jetbrains.python.PythonLanguage;
 import org.jetbrains.annotations.NotNull;
 
@@ -81,4 +82,15 @@ public class RestFileViewProvider extends MultiplePsiFilesPerDocumentFileViewPro
     }
     return null;
   }
+
+  @Override
+  public IElementType getContentElementType(@NotNull Language language) {
+    if (language == getTemplateDataLanguage()) {
+      return RestPythonElementTypes.PYTHON_BLOCK_DATA;
+    }
+    else if (language.getID().equals("DjangoTemplate")) {
+      return RestPythonElementTypes.DJANGO_BLOCK_DATA;
+    }
+    return null;
+  }
 }